1. 查看自己的CUDA版本,打开cmd:
nvcc --version
所以CUDA 版本为10.2。
2. 打开 pytorch官网 ,选择适合自己的安装命令:
以管理员身份运行!以管理员身份运行!Anaconda Prompt !(不然,可能报错)
(即使报错,只要你不断地重新输入命令,直到它可以安装为止!因为有可能是网络问题。)
切到自己所要安装pytorch的环境,输入以上Command:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
当然,也可以下载以前的Pytorch版本:https://pytorch.org/get-started/previous-versions/
————————耐心等待下载————————
3. 验证是否下载成功:
- 输入python
- 输入 import torch
- 输入 torch.cuda.is_available() 查看是否能调用gpu,没有报错即安装成功。
参考链接: