Pytorch安装踩坑:
个人学习时踩坑的经验,仅作为记录。
目前已经通过pip离线安装解决了GPU版本的安装问题(8.4更新)
1.安装GPU Torch
1.1 官网在线安装
1.查看本机的CUDA版本nvidia-smi
,得知支持的版本为cuda11.6,在官网链接: pytorch历史版本中查到对了对应版本的安装命令,如下
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia
不出意外的报网络错误
1.2 清华源在线安装
参考了