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即为环境搭建成功!

相关推荐
蹦蹦跳跳真可爱58926 分钟前
Python----OpenCV(图像増强——高通滤波(索贝尔算子、沙尔算子、拉普拉斯算子),图像浮雕与特效处理)
人工智能·python·opencv·计算机视觉
nananaij30 分钟前
【Python进阶篇 面向对象程序设计(3) 继承】
开发语言·python·神经网络·pycharm
雷羿 LexChien36 分钟前
从 Prompt 管理到人格稳定:探索 Cursor AI 编辑器如何赋能 Prompt 工程与人格风格设计(上)
人工智能·python·llm·编辑器·prompt
qqxhb39 分钟前
零基础数据结构与算法——第四章:基础算法-排序(上)
java·数据结构·算法·冒泡·插入·选择
敲键盘的小夜猫1 小时前
LLM复杂记忆存储-多会话隔离案例实战
人工智能·python·langchain
高压锅_12202 小时前
Django Channels WebSocket实时通信实战:从聊天功能到消息推送
python·websocket·django
FirstFrost --sy2 小时前
数据结构之二叉树
c语言·数据结构·c++·算法·链表·深度优先·广度优先
森焱森3 小时前
垂起固定翼无人机介绍
c语言·单片机·算法·架构·无人机
槑槑紫3 小时前
深度学习pytorch整体流程
人工智能·pytorch·深度学习
搂鱼1145143 小时前
(倍增)洛谷 P1613 跑路/P4155 国旗计划
算法