安装CUDA后安装PyTorch，pip install torch 太慢【解决办法】 实测可行torch.cuda.is_available() return True

最近学习深度学习，训练模型时发现用CPU太慢，恰好电脑有一张NVIDIA的1060显卡，所以就打算下载NVIDIA的运算平台CUDA来使用GPU加快训练了。

我的python版本是3.7，装完CUDA10.2后，接下来就是安装pytroch了。由于我安装的是最新的稳定版，所以使用conda安装发生了找不到Package的问题，于是我用pip安装

 

在命令行输入

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，大功告成！