Anaconda下面安装pytorch
pytorch CPU版本的安装
1.进入官网https://pytorch.org/,找到适合自己的pytorch版本,我的选择版本如下:
复制"Run this Command"的两条命令。pip install torch1.7.0+cpu torchvision0.8.1+cpu torchaudio=0.7.0 -f https://download.pytorch.org/whl/torch_stable.html。,如果直接运行上面代码,由于安装过程需要从国外的服务器下载安装包,大概几百兆的样子,速度可能会很慢很慢很慢…。为了解决下载速度慢的问题,可以在上面的所有命令后面加上: -i https://pypi.tuna.tsinghua.edu.cn/simple。一起复制到cmd命令窗口中,我用的是ANaconda Prompt命令窗口,命令如下:
pip install torch1.7.0+cpu torchvision0.8.1+cpu torchaudio=0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple。
在下载过程中,第一次出现下载失败错误,重新试了一次,非常快就下载完了。
3.进入python进行测试
import torch
print(torch.version)
import torchvision
print(torchvision.version)
运行这四行代码,没有出错说明安装成功。