我是在(windows)anaconda虚拟环境中下载pytorch:
首先找到下载pytorch的命令, https://pytorch.org/ 在这个网站中有下载命令
建议使用pip进行下载,conda命令下载经常出现莫名其妙错误(我经常出错)
cuda版本如何看呢?(安装如果用tensorflow-gpu的话,每个tensorflow-gpu对应不同的cuda版本,选择最适合自己的)
(1)桌面右键选择:NVIDIA控制面板
(2)选择 帮助 中 系统信息
(3)点击 组件
下载后进行导入import torch可能出现以下错误:(无错请忽略)
莫慌!按照提示的网站在浏览器打开进行下载安装:我的是 https://aka.ms/vs/16/release/vc_redist.x64.exe
再次import torch 可能你还会出现这个错误:(无错请忽略)
莫慌!大概率是numpy版本不对,先把自己的numpy卸掉:
pip uninstall numpy
建议安装numpy-1.15.0版本,我用豆瓣源进行下载(推荐)
pip install -i https://pypi.douban.com/simple numpy==1.15.0
再次试一试:霍~这不就好了莫.....