快速安装torch的方法
正常安装torch是直接使用pip install pytorch xxxxx
但是很有可能是安装时间很长大概一两个小时
那么原因可能是你的Python版本和pytorch版本不太匹配
提供一个快捷方法:下载.wheel:
首先确定一下自己cuda的版本若cuda版本跟安装的torch版本不匹配那么torch.cuda.is_available()就会返回false,也就是说torch还是无法正常使用,在cmd使用指令:
nvidia-smi
就可以查看电脑cuda版本。
torch下载地址:download.pytorch.org/whl/torch_stable.html
下载torch,当然打开梯子的话下载的更快
然后下载torchaudio
然后下载:torchvision
或者在cmd使用指令:
pip install +地址(右键复制)
可以下载的更快。