|
1.首先下载anaconda软件
2.在anaconda中创建一个环境：conda create -n pytorch python=3.6(在anaconda中创建了一个名为pytorch的环境，该环境的python版本为3.6)
3.检查自己的显卡类型（任务管理器-性能-GTX1650）以及驱动版本（在anaconda prompt中输入指令nvidia-smi）,本文的nvidia-smi和driver version=457.49（不同的cuda版本有最低的驱动版本要求），cuda version=11.1（最高支持的版本？）。
4.在https://www.geforce.com/hardware/technology/cuda/supported-gpus中查看自己的显卡是否支持cuda
5.https://pytorch.org/get-started/locally/下载符合的cuda版本，例如在anaconda中的pytorch环境中下载cuda=10.2的指令conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch（该网址中显示的可安装版本可能不同的机器会显示不同：和驱动版本和显卡有关？我显示的是10.2和11.3）
6.遇到的问题：两台同样都是GTX1650的机器，一台cuda版本是10.0可正常使用，但是另一台不行，cuda版本换成了上文提到的网站中直接显示的10.2之后torch.cuda.is_available()就变成了true
|