1.首先,网上有提到运用虚拟环境去装pytorch,为啥要虚拟环境?
在从github上面下载别人的代码之后,不同的代码往往需要特定的运行环境。比如说有些代码需要在python3.6的环境下运行,有些代码需要在python2.7的环境下运行。
这就到了anaconda大显身手的时候了。Anaconda主要是对python中的各个包进行管理与部署。
所以,目前也可以不用在虚拟环境下安装,等腰用到其他的python版本的环境时再建也可以。
2.接着,去pytorch的官网选择需要安装的版本,找到对应的指令,比如,我用的是台式机只有CPU,就选择了CUDA为None的,如果有GPU就选择对应的。
现在,打开可以运行指令的终端,比如可以在pycharm中的Terminal。
pycharm挺好用的编译器,如果没有安装,可以参考
https://blog.csdn.net/weixin_42915878/article/details/102933136
3. 这里我们不用官网安装pytorch,因为速度慢。
输入以下指令进行安装,配置镜像源之后安装快一点 所以用这条命令
加入清华镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
可以直接复制,输入后回车。
最后,再输入第2步的指令
conda install pytorch torchvision torchaudio cpuonly
注:这里要去掉-c pytorch
输入以上命令即可成功安装
当你可以看到以下的内容,说明已经装好了,可以试着敲代码验证了。
4.检测pytorch是否安装成功
import torch
import torchvision
x = torch.rand(5,5)
print(x)
print(torch.__version__)
如果有输出,则Pytorch安装成功
5.如果是第3步正常,但是第4步异常
输出错误,首先看一下编译器设置有没有设置对,选择python要选择anaconda的pyhon,如下图:
如果不是,点击show all 那里,找到anaconda的安装位置,找到目录下的python
如果是另外的错误,就需要看具体的错误是什么再对应的解决哦。