Windows系统下CUDA、CuDNN、Pytorch的安装、卸载以及安装过程中部分问题的解决

1、安装背景

需要使用Python做一些深度学习上的应用,已安装Python、Anaconda、2017版VS、2019版VS

(我要安装的CUDA版本是12版本往后的版本,部分操作可能与12版本之前的安装操作不同)

2、CUDA的安装

首先需要确定自己电脑能安装哪个版本的CUDA

通过英伟达控制面板查看能安装CUDA的最高版本

然后根据版本去CUDA官网点击对应的下载链接

CUDA下载官网

这里一般选择本地下载(local),下载好后,运行exe,具体步骤可看下面这篇文章的第二部分:

CUDA安装详细步骤

3、CuDNN的下载

CuDNN版本要与CUDA版本对应,这里我CUDA-12.4版本选用的是CuDNN-9.1.0版本,CUDA-12.3版本选用的是CuDNN-8.9.7版本

以下是CuDNN的官方下载地址:

CuDNN官网下载地址

4、CuDNN安装以及相关环境配置

可参考以下文章进行CuDNN安装以及环境的配置:

环境配置

5、Pytorch安装

Pytorch的版本要注意与CUDA版本对应,可以去官网复制粘贴安装命令,并在Anaconda中的目标虚拟环境中运行:

Pytorch下载

在Pytorch官网里面有让你选择Compute Platform的选项,一般只有CUDA12.1和12.4的选项,实际上,如果你的CUDA是12.3版本的,那么你可以选择Compute Platform为12.1

6、安装问题(待补充)

问题一:因为Nsight Compute安装失败而导致的CUDA安装失败

这个问题我在CUDA12.4版本安装中遇到过,解决办法就是单独下载这个模块,然后再运行CUDA的安装程序,在自定义安装中把把模块取消掉(因为已经下载过了),具体操作步骤可看下面这篇文章:

Nsight Compute安装失败的一个方法

7、卸载

有的时候会因为版本冲突的原因,需要对CUDA或者Pytorch更换版本,下面简单介绍如何对它们进行卸载重装。

CUDA卸载及重装

在“控制面板中”点击“卸载程序”,然后搜素“cuda”,将相关程序卸载即可

重装的时候在CUDA的安装运行程序中,在自定义安装里面,只需勾选“CUDA”即可

Pytorch卸载

在Anaconda的目标虚拟环境中运行

pip uninstall torch

  • 8
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值