最近学习深度学习,训练模型时发现用CPU太慢,恰好电脑有一张NVIDIA的1060显卡,所以就打算下载NVIDIA的运算平台CUDA来使用GPU加快训练了。
我的python版本是3.7,装完CUDA10.2后,接下来就是安装pytroch了。由于我安装的是最新的稳定版,所以使用conda安装发生了找不到Package的问题,于是我用pip安装
https://pytorch.org/get-started/locally/ (官方安装指南)
在命令行输入
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
发现速度极慢,几十KB/S,而且中间可能会发生短连,于是我打算手动下载.whl安装
访问下面的地址
找到对应版本的torch=1.4和torchvision=0.6下载,我这里用IDM下载,几分钟就下好了
然后到下载好的.whl路径下,打开命令行cmd
输入pip install xx.whl
接下来就是测试Torch能不能用CUDA了
True,大功告成!