cuda+cudnn+pytorch2.2安装

部署运行你感兴趣的模型镜像

之前安装GPU一直是囫囵吞枣,秉承能用就行的原则,现在尝到了恶果,从这次开始记录电脑的配置环境

由于要复现一篇文章代码,他用的torch=2.2,同时由于复现代码导致突然原来的cuda报错OSError: [WinError 127] 找不到指定的程序。 Error loading “D:\Anaconda3\envs\pytorch39_cuda\lib\site-packages\torch\lib\c10_cuda.dll” or one of its dependencies.

尝试多种方法解决未果
因此,重新用anoconda创建环境,重新配置

声明情况: 目前电脑已经有cuda11.0版本和10.0版本

主要参考:文章配置环境

1. 输入nvidia-smi

从右上角可以看出工位电脑cuda支持的最高版本是12.4
在这里插入图片描述由于复现的代码torch为2.2,经过网上搜索发现cuda11.8+torch2.2可以支持,因此以下步骤全部参考自https://blog.youkuaiyun.com/imok1234567/article/details/140251769

2. 下载cuda11.8

进入下载页面CUDA Toolkit Archive | NVIDIA Developer(https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local),选择cuda11.8,这个版本用的人很多,可以支持绝大部分的深度学习项目。
根据自己电脑系统情况选择下面的选项,最终点击download
在这里插入图片描述
双击 .exe 文件
更改 CUDA 安装时缓存位置(默认为 C盘,不更改也可以,但得与原先安装的 CUDA 的安装缓存位置不同,或者该位置为空)下面参考的这篇文章
在这里插入图片描述我是直接把C改为了D

同意软件许可协议并继续

选择自定义安装

勾选驱动程序组件
由于之前已经安装过 CUDA,此时我们只需要选择 CUDA 即可
(这里我的与参考链接中不一致,我的有四个选项,所以把后两个取消了勾选)
在这里插入图片描述
如果之前没有安装过vs,则需要把CUDA里面的
Visual Studio Integration取消勾选,否则会安装不成功
在这里插入图片描述

选择安装位置
默认安装位置为 C 盘,可以自由选择安装到其他盘,但需要记住安装位置,后期会使用到(我的只有后两个)
在这里插入图片描述选择 I understand,并继续

安装结束在这里插入图片描述查看环境变量是否配置

搜索查看高级系统设置->高级-> 环境变量
可以看到有这两项
在这里插入图片描述
验证安装是否成功
在这里插入图片描述# 3. 下载cudnn
进入下载页面cuDNN 历史版本 | NVIDIA 开发者,选择”下载 cuDNN v8.9.7 (2023年12月5日), 适用于 CUDA 11.x“,后面的适用于CUDA 11.x意思是支持cuda11的任何一个版本,也包括本次选择的cuda11.8。然后点击”Local Installer for Windows (Zip)“,这里需要注册用户才能下载,由于是选择了英伟达在中国的官网,所以下载速度会快一点。

没登陆英伟达网站的话,过程中会跳转需要登录,登录后再重新进入,依次点击下面红色
在这里插入图片描述解压下载好的cudnn安装包,同时打开刚刚安装cuda的地址,如果是默认路径, 那么是在”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8“路径下。

整体cudnn的安装过程是将右边cudnn文件夹下的文件,复制到左边的cuda目录下。首先将cudnn文件夹中的bin和include两个文件夹中的内容,拷贝到cuda安装目录下的bin和include文件夹中。在cudnn中的lib文件夹下,有一个x64文件夹,同理将里面的内容,拷贝到cuda安装目录下的lib文件夹下的x64文件夹中。
在这里插入图片描述

4.创建虚拟环境

输入代码创建虚拟环境,其中“ai”,替换为自己想创建的虚拟环境名字。

conda create -n ai python==3.10
这里我创建的是

conda create -n cuda118cudnn89torch22py39 python==3.9

5.torch环境配置

进入pytorch官网Previous PyTorch Versions | PyTorch,选择torch2.2.2—CUDA 11.8的代码
conda install pytorch2.2.2 torchvision0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
在这里插入图片描述耐心等待

后来报错内存不够,所以音频的没装
改为下面的代码(先不要安,后面报错了):
pip install torch2.2.2+cu118 torchvision0.17.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

报错:

Installing collected packages: numpy, networkx, torch, torchvision
ERROR: pip’s dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
torchmetrics 1.4.0.post0 requires packaging>17.1, which is not installed.
Successfully installed networkx-3.2.1 numpy-2.0.2 torch-2.2.2+cu118 torchvision-0.17.2+cu118

解决办法:
先conda install packaging
然后再pip install torch2.2.2+cu118 torchvision0.17.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
验证安装成功
在这里插入图片描述

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

<think>好的,我现在需要帮助用户解决在Linux系统上安装CUDAcuDNNPyTorch的问题,同时要注意版本兼容性。首先,我需要回顾用户提供的引用内容,确保信息准确。 用户提到参考了安装CUDA之前的步骤,比如确认GPU是否支持CUDA,访问NVIDIA官网下载对应版本,安装过程包括驱动程序和库的选择,以及验证安装。还有Anaconda和虚拟环境的搭建,PyTorch安装。这些步骤需要详细梳理。 首先,安装前的准备:检查GPU型号,确保支持CUDA。用户可能在Linux下使用lspci或nvidia-smi命令。然后安装NVIDIA驱动,可能需要添加官方仓库或使用.run文件手动安装。这部分需要注意驱动版本与CUDA Toolkit的兼容性。 接下来是CUDA Toolkit的安装步骤,根据引用中的说明,下载对应版本的.run文件,执行安装安装时可能需关闭图形界面,进入命令行模式。然后配置环境变量,如PATH和LD_LIBRARY_PATH,这一步很重要,否则系统可能找不到CUDA库。 然后是cuDNN安装,用户需要注册NVIDIA开发者账号,下载对应CUDA版本的cuDNN,解压并复制文件到CUDA目录。需要确保cuDNN版本与CUDA版本匹配,否则可能出错。 接下来是Anaconda安装,用于创建虚拟环境,管理Python依赖。安装后创建环境,激活,然后安装PyTorchPyTorch官网提供了不同CUDA版本的安装命令,必须选择与已安装CUDA版本兼容的PyTorch版本,否则无法利用GPU加速。 版本兼容性方面,需要确认CUDA Toolkit、cuDNNPyTorch之间的对应关系。例如,PyTorch 1.8可能要求CUDA 11.1,而cuDNN需要对应CUDA 11.x的版本。用户需要查阅各官方文档的版本要求,避免版本冲突。 验证安装部分,CUDA可以通过编译sample并运行deviceQuery来检查。cuDNN则可能通过检查头文件和库文件,或者运行测试代码。PyTorch中则用torch.cuda.is_available()确认是否识别GPU。 可能遇到的问题包括驱动安装失败,CUDA路径未正确设置,版本不匹配导致运行时错误。需要提醒用户仔细检查版本,参考官方文档,逐步操作。 现在需要将这些步骤整理成清晰的教程,分步骤说明,强调版本兼容性,并给出验证方法。同时生成相关问题,帮助用户深入理解可能遇到的后续问题。</think>以下是在Linux系统上安装CUDAcuDNNPyTorch的详细步骤及版本兼容性说明: --- ### **1. 安装前准备** #### **1.1 确认GPU支持CUDA** - 执行命令查看GPU型号: ```bash lspci | grep -i nvidia ``` - 访问[NVIDIA CUDA支持列表](https://developer.nvidia.com/cuda-gpus)确认是否支持[^2]。 #### **1.2 安装NVIDIA驱动** - 禁用系统默认驱动: ```bash sudo nano /etc/modprobe.d/blacklist-nouveau.conf # 添加内容:blacklist nouveau / options nouveau modeset=0 sudo update-initramfs -u ``` - 重启后安装驱动: ```bash sudo apt install nvidia-driver-535 # 根据CUDA版本选择驱动版本 ``` --- ### **2. 安装CUDA Toolkit** #### **2.1 下载CUDA** - 访问[CUDA Toolkit官网](https://developer.nvidia.com/cuda-toolkit-archive),选择与驱动兼容的版本(例如CUDA 11.8)。 #### **2.2 安装CUDA** - 执行安装命令(以CUDA 11.8为例): ```bash sudo sh cuda_11.8.0_520.61.05_linux.run ``` - **安装选项**:取消勾选驱动(已提前安装)和所有组件(默认全选)。 #### **2.3 配置环境变量** - 在`~/.bashrc`中添加: ```bash export PATH=/usr/local/cuda-11.8/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH ``` - 生效配置: ```bash source ~/.bashrc ``` #### **2.4 验证CUDA** - 编译并运行示例程序: ```bash cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make && ./deviceQuery ``` - 输出显示`Result = PASS`表示成功。 --- ### **3. 安装cuDNN** #### **3.1 下载cuDNN** - 访问[cuDNN官网](https://developer.nvidia.com/cudnn),选择与CUDA版本兼容的cuDNN(例如CUDA 11.x对应cuDNN 8.9.4)。 #### **3.2 安装cuDNN** - 解压并复制文件到CUDA目录: ```bash tar -xzvf cudnn-linux-x86_64-8.9.4.25_cuda11-archive.tar.xz sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* ``` --- ### **4. 安装PyTorch** #### **4.1 创建Anaconda虚拟环境** - 安装Anaconda后创建环境: ```bash conda create -n pytorch_env python=3.9 conda activate pytorch_env ``` #### **4.2 安装PyTorch** - 根据CUDA版本选择PyTorch命令(CUDA 11.8对应PyTorch 2.0+): ```bash pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 ``` #### **4.3 验证PyTorch** - 运行Python检查GPU支持: ```python import torch print(torch.__version__) # 输出版本号(如2.0.1) print(torch.cuda.is_available()) # 输出True表示成功 ``` --- ### **版本兼容性表** | 组件 | 推荐版本 | 兼容范围 | |------------|----------------|------------------| | CUDA | 11.8 | 11.0~11.8 | | cuDNN | 8.9.4 | 需匹配CUDA版本 | | PyTorch | 2.0.1 | 需指定`cu118` | | NVIDIA驱动 | ≥535.86.05 | 需支持CUDA 11.8 | --- ### **相关问题** 1. 如何解决安装CUDA时出现的驱动冲突问题? 2. cuDNN未正确安装会导致哪些PyTorch错误? 3. 如何在多版本CUDA之间切换? 4. PyTorch如何强制使用CPU而非GPU? --- **引用说明** - 环境变量配置和驱动安装步骤参考了CUDA官方文档。 - PyTorch安装命令需严格匹配CUDA版本[^1]。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值