通过虚拟环境virtualenv安装pytorch

首先提及,下面的命令,我都是直接在macos上的pycharm中的terminal执行的

1、安装virtualenv虚拟环境:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  --upgrade virtualenv

或者 python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  --upgrade virtualenv

2、创建虚拟环境

  virtualenv --system-site-packages -p python3 targetDirectory

 注意需要找到virtualenv 安装的位置,然后targetDirectory是你想要存放虚拟环境的路径,下面是我的使用,我自己在pycharm的项目路径下,新建了venv目录并在该目录下为我即将创建的tensorflow环境,建了子目录,以后就在这个venv下,管理我的所有虚拟环境

 ./virtualenv --system-site-packages -p python3 ~/PycharmProjects/venv/pytorch01/

3、激活虚拟环境

source ~/PycharmProjects/venv/pytorch01/bin/activate 

显示如下图

4、在激活的虚拟环境中安装 Pytorch 及其所有依赖

pip install torch torchvision
鉴于上面的命令实在太慢了,耗时长,我用了清华的镜像,如下

 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  torch torchvision

这样发现可能报错,可以换个豆瓣的源试试

pip install torch torchvision -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

其中--trusted-host pypi.douban.com 的作用是获得ssl证书的认证,这个可以先不加,不行的话再加上

同理,清华的也可以加上--trusted-host 试一试

然后版本记得匹配就好,我用的是torch==1.6.0 torchvision==0.7.0

5、验证安装效果

在命令行输入python,进入python环境,import pytorch 如果没有报错,就安装ok

下面的简单语句,可以测试一下输出

import torch

print(torch.rand(2,3))

恭喜各位开启崭新的pytorch之旅吧。如果安装有问题,欢迎随时交流哦。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐