由于大创需要机器学习的框架pytorch,安装过程中这里写一下下载配置过程中遇到的坑。
cpu版本:
只需要复制官网中自己对应版本的pip下载指令就可以了,网速还是很快的
然而下载后才发现代码是需要gpu的
在控制面板中查看自己的cuda版本
(1)按win+Q,输入控制面板,然后查看方式小图标,点击NVIDIA控制面板:
点击NVIDIA控制面板的帮助,点击左下角系统信息:
点击组件:这里就显示了你的CUDA的信息啦!!!
[图片源自 https://blog.csdn.net/qq_38295511/article/details/89223169 ]
我的CUDA版本是 9.1.84,python版本3.6.2 win10
使用如下语句可以安装cu91版本的pytorch
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
然而下载torchvision会使用之前缓存中的torchvision而且0.4.1版本的并不能使用
最后使用
pip install torchvision==0.1.8
成功安装