pytorch-gpu安装注意事项

245 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

1. CUDA与cuDNN安装

1.1 GPU驱动版本检查

win+R输入cmd,回车打开。输入nvidia-smi查看当前驱动的版本号,版本数值可通过更新驱动增加。 在这里插入图片描述

1.2 CUDA版本选择

按照驱动版本与表中对应的CUDA版本选择合适的版本号。docs.nvidia.com/cuda/cuda-t… 如我的驱动版本为511.65,我之后安装了CUDA 10.2 在这里插入图片描述 CUDA各版本下载地址:developer.nvidia.com/cuda-toolki…

1.3 CUDA与cuDNN安装教程

CUDA安装官方教程:docs.nvidia.com/cuda/cuda-i… 详细的安装教程:blog.csdn.net/weixin_4384…,记得配置环境变量

2. 安装Anaconda

3. 安装Pytorch

在这里插入图片描述 还是用上面的安装教程:blog.csdn.net/MCYZSF/arti…,一般来说时间比较久,我爬着某科学软件,所以十分钟左右安装好了。 最后按照教程里的代码去检验安装即可。

  • 在其它很多教程里有换了清华源的,但是好像清华源没有版权停止了更新,我尝试了很多种教程中不同的清华源配置都无法安装Pytorch,所以最后我没有更改源。
  • 不过也有说源已经恢复了mirrors.tuna.tsinghua.edu.cn/news/anacon…

4. 在Pytorch中安装Jupyetr notebook

直接翻到该安装教程的最后一步:cloud.tencent.com/developer/a…

5.结语

中间遇到了很多报错,如

等等