cuda 11情况下如何配置pytorch 10.2

最新推荐文章于 2024-04-01 22:29:44 发布

加油上学人

最新推荐文章于 2024-04-01 22:29:44 发布

阅读量9.5k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：问题解决 pytorch Ubuntu 文章标签： cuda 11 pytorch 10.2 cuda toolkit cuda降低版本

本文链接：https://blog.youkuaiyun.com/weixin_45885232/article/details/109092412

问题解决同时被 3 个专栏收录

65 篇文章

订阅专栏

pytorch

13 篇文章

订阅专栏

Ubuntu

13 篇文章

订阅专栏

该博客指导如何在Ubuntu系统中将CUDA从11.0降级到10.2，以适应PyTorch 10.2版本的需求。首先卸载CUDA 11.1，然后下载并安装CUDA 10.2及驱动。接着，安装cudnn并测试CUDA是否正常工作。最后，通过官方渠道下载并安装PyTorch，验证其GPU可用性。整个过程详述了每个步骤，确保PyTorch能在CUDA 10.2环境下顺利运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于目前pytorch只能支持到10.2的版本，但ubuntu最新的系统驱动直接支持了cuda 11.0，并且cuda tooklit支持的默认下载也是11.0。

1. 需要先降低cuda tooklit的版本

cuda-uninstaller in /usr/local/cuda-11.1/bin
sudo rm -rf /usr/local/cuda-11.1

来自：Nidia
或

cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.1

来自：NVIDIA CUDA Toolkit 11.0 安装与卸载(Linux/Ubuntu)

2. 之后下载安装cuda 10.2:

下载地址：
CUDA Toolkit 10.2 Download

在这里插入图片描述

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.runsudo sh cuda_10.2.89_440.33.01_linux.run

3. 只安装驱动外的其他内容，安装结束后测试是否成功：

cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

出现PASS，代表没有问题

4. 安装cudnn
去官网选择适合自己的版本，有两种可选：
在这里插入图片描述和

此处我选择的第二个，下载这三个文件：

在这里插入图片描述
之后使用命令安装：

sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

5. 安装pytorch
官网选择即可，如果网速慢可参考另一篇文档：
在这里插入图片描述
网速比较快，出现下面的内容说明配置正确：

6. 测试
输入命令：

python
>>> import torch
>>> torch.cuda.is_available()

在这里插入图片描述
完结撒花！

参考：
NVIDIA CUDA Toolkit 11.0 安装与卸载(Linux/Ubuntu)
pytorch
cuDNN Archive
CUDA Toolkit 10.2 Download
Ubuntu18.04下安装pytorch详细步骤