Windows 11在conda虚拟环境中安装cuda、cudann

前言:此篇文章以Pytorch2.0.0+CUDA11.8+cuDNN8.9.2为例

1、为什么要在虚拟环境中安装cuda、cudann?

(1) 隔离依赖

虚拟环境可以帮助隔离不同项目所需的依赖库和版本,避免冲突。例如,不同项目可能需要不同版本的 CUDA 或 cuDNN 来支持特定的深度学习框架(如 TensorFlow、PyTorch)的版本要求。将这些库安装在虚拟环境中,确保每个项目使用其特定的版本,避免系统环境中的全局冲突。

(2)方便管理和维护

在虚拟环境中安装 CUDA 和 cuDNN 可以简化管理和更新流程。例如,针对某个项目进行版本升级或降级时,可以在虚拟环境内更改依赖而不会影响到其他项目的运行。

(3)提升兼容性

在深度学习开发中,不同的深度学习框架(如 TensorFlow 和 PyTorch)可能对 CUDA、cuDNN 的版本有不同的要求。在虚拟环境中安装专用版本,能够确保不同项目在不同的依赖版本下正确运行,避免在系统范围内安装多个版本而导致冲突。

(4)方便迁移和部署

使用虚拟环境将 CUDA、cuDNN 和项目代码、库等绑定在一起,便于在不同的机器上进行迁移或部署。这样可以确保在另一台机器上运行时,环境一致,减少因环境差异带来的问题。

(5)快速重置环境

在开发和测试中,可能需要快速重置或重建环境。将 CUDA 和 cuDNN 安装在虚拟环境中,可以更方便地删除或重建,而不会影响到全局的系统配置。

2、虚拟环境

(1)创建conda虚拟环境

conda create -n name python=3.x

python 复制代码
conda create -n chatGLM python=3.9

(2)激活conda虚拟环境

conda activate name

python 复制代码
conda activate chatGLM

ps:后面的操作均在激活的虚拟环境执行

(3)查看CUDA版本

系统的 CUDA 驱动版本决定了支持的最高 CUDA Toolkit 版本,并且该驱动对低版本的 CUDA Toolkit 具有向下兼容性。

python 复制代码
nvidia-smi

比如我的CUDA Version=12.7,那么它就可以支持 ≤ 12.7版本的CUDATookit

(4)安装cudatoolkit和cudnn包

查看支持的cuda版本

python 复制代码
conda search cudatoolkit
或者
conda search cudatoolkit --info

进入pytorch官网:网址

复制

python 复制代码
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia

出现done完成!如果出现以下情况,再试一次;

安装特定版本的cuda:

python 复制代码
conda install cuda -c nvidia/label/cuda-11.8.0
conda install cuda-cudart-dev -c nvidia
或者
conda install nvidia/label/cuda-11.8.0::cuda --channel nvidia/label/cuda-11.8.0

如果出现:

python 复制代码
conda install -c nvidia cuda-nvcc=11.8.89 --clobber

安装指定版本的cudnn:

python 复制代码
conda install cudnn=8.9.2

pip list查看是否安装完成

验证是否安装成功:

python 复制代码
python
import torch
torch.cuda.is_available()

出现true即为环境搭建成功!

相关推荐
梓仁沐白8 分钟前
ubuntu+windows双系统切换后蓝牙设备无法连接
windows·ubuntu
网易独家音乐人Mike Zhou2 小时前
【卡尔曼滤波】数据预测Prediction观测器的理论推导及应用 C语言、Python实现(Kalman Filter)
c语言·python·单片机·物联网·算法·嵌入式·iot
安静读书2 小时前
Python解析视频FPS(帧率)、分辨率信息
python·opencv·音视频
小二·3 小时前
java基础面试题笔记(基础篇)
java·笔记·python
九鼎科技-Leo4 小时前
什么是 WPF 中的依赖属性?有什么作用?
windows·c#·.net·wpf
小喵要摸鱼5 小时前
Python 神经网络项目常用语法
python
Swift社区5 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Kent_J_Truman6 小时前
greater<>() 、less<>()及运算符 < 重载在排序和堆中的使用
算法
一念之坤6 小时前
零基础学Python之数据结构 -- 01篇
数据结构·python
IT 青年6 小时前
数据结构 (1)基本概念和术语
数据结构·算法