Windows10下CUDA, cuDNN, pytorch, torchvision的安装过程(一）----CUDA,cuDNN安装

最新推荐文章于 2024-07-25 17:01:46 发布

困坤的小菜鼠

最新推荐文章于 2024-07-25 17:01:46 发布

阅读量1.4k

点赞数 2

分类专栏： CUDA 文章标签： windows cuda pytorch

本文链接：https://blog.csdn.net/weixin_45654744/article/details/110947834

版权

本文详述了在Windows10上安装CUDA10.2、cuDNN以及PyTorch的过程，包括解决安装CUDA时遇到的版本匹配、环境变量配置等问题。在安装过程中，强调了CUDA与Python、PyTorch的版本兼容性，以及安装后的环境变量设置和验证步骤。同时，还提供了CUDNN的下载和安装指南。

摘要由CSDN通过智能技术生成

前言：

我的Windows10在安装CUDA, cuDNN, pytorch，torchvision的过程中踩过几次坑，反复尝试了挺多次的，在网上也找了好多的教程，最后终于摸索出了方法，所以想记录下来，以便之后可以回过头来参考。

作为小白，我在安装的过程中主要遇到了这几种类型的坑：

我的电脑已经安装了Anaconda， python3.8，不知道CUDA的版本和python的版本也是需要适配的；
安装了CUDA8.0，但是不知道怎么配置环境变量；
不知道DUAD和pytorch, torchvision, python的版本都是需要适配才行；
电脑同时安装了CUDA8.0和CUDA10.2,安装过程中遇到了“有安装程序正在运行，请关闭安装程序后再进行新的安装/卸载”
安装好CUDA, pytorch, torchvision之后，在python环境中可以import, 但是在pycharm中不能import torch和torchvision。

接下来我会将CUAD10.2, cuDNN, pytorch, torchvision的安装过程记录下来，并将踩到的坑的解决过程记录下来。

安装前版本匹配建议： CUAD 10.2 ---- pytorch 1.6.0 ----torchvision 0.7.0
在这里插入图片描述
这是我在pytorch下载官网找到的，如果你要寻找其他版本的匹配模式的话，可以参考这个网站 https://pytorch.org/get-started/previous-versions/

安装前注意点：我的电脑在安装CUDA10.2之前已经安装了Anaconda3, python3, CUDA8.0, 而CUDA8.0版本太低，与python3.8不适配，后面可能对于其他工具的安装版本也有影响，所以我打算重新安装CUDA10.2。我可以选择先将CUDA8.0卸载再安装CUDA10.2, 也可以选择CUAD10.2和CUDA8.0并存。CUDA安装方式并未受影响，都是同样的安装方式。