安裝 torch1.7#
1. 安裝好 cuda 和 cudnn#
具體可看我之前發布的文章
2. 在官網找到適合自己的版本#
Previous PyTorch Versions | PyTorch
我安裝的是 cuda11 windows 系統 使用這個命令 不同的系統在這裡的安裝命令都不太一樣，具體看官方的代碼
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
沒有報錯就是安裝完畢了，如果安裝很慢請換 pip 的源