安装:Anaconda3
通过命令行查显卡nvidia-smi
打开Anacanda prompt 新建 conda create -n pytorch python=3.6
在Previous PyTorch Versions | PyTorch选择1.70,安装成功,但torch.cuda.is_available 返回false
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=9.2 -c pytorch
选择1.60,安装成功,但torch.cuda.is_available 返回false
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch
选择1.5.0,安装失败,torchvision=0.6.0 不合适
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=9.2 -c pytorch
最后选择1.5.1,安装成功,torch.cuda.is_available 返回true
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch