Pytorch 安装 GPU版本

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. 验证是否下载成功:

  1. 输入python
  2. 输入 import torch
  3. 输入 torch.cuda.is_available() 查看是否能调用gpu,没有报错即安装成功。


参考链接:

pytorch安装最容易实现版本,踩坑无数总结

猜你喜欢

转载自blog.csdn.net/baidu_30506559/article/details/121912163