最近在win10上使用pytorch,涉及到了安装的一些问题,在这里整理出来,帮助小伙伴们少踩点坑。
首先安装anaconda,这点不再赘述,进入conda选择需要安装的env,开始安装;
进入torch官网,https://pytorch.org/get-started/locally/ ,这里可以根据你的机器配置推荐合适的安装命令,如图
前面几行的选项基本上默认就可以(除非使用C++/java版的pytorch),CUDA表示GPU的相关设置,如果不使用GPU选择None,选择完毕会生成相应的建议命令行。这里需要注意的一点是,生成的命令行中 -c pytorch 表示conda获取pytorch的channel,使用conda自带的pytorch仓库,这种情况下即使conda使用了清华等国内源下载速度也是超级慢,所以要在安装时去掉 -c pytorch,这样就能很快下载安装完成。
正常情况下使用conda都可以顺利安装好,如果实在下载不动,可以考虑下载pytorch离线.whl文件,这里提供一个下载的链接,具体安装步骤不再赘述。
进入python,运行以下命令,如果没有报错,并正常输出版本号,则说明安装成功。
>>>import torch
>>>import torchvision
>>>print(torch.__version__)
1.6.0