ubuntu20.04安装cuda和cudnn

本文档详细记录了在Ubuntu20.04上安装英伟达驱动515.65.01、CUDA11.0.2、cuDNN8.0.5的过程,以及通过conda安装Pytorch、torchvision和torchaudio,特别提到了cudatoolkit=11.3的指定版本安装。同时,提供了卸载Pytorch的pip和conda方法。
摘要由CSDN通过智能技术生成

以下安装的版本

1、英伟达驱动:515.65.01

2、CUDA:11.0.2

3、CUDNN: 8.0.5.29

4、pytorch安装的方式,在官网选择命令行输入

conda install pytorch torchvision torchaudio cudatoolkit=11.3

下载英伟达驱动

(43条消息) Ubuntu20.04安装英伟达驱动步骤_Rootzzy1216的博客-CSDN博客

下载cuda

1、在官网查看合适的版本

我此处下载11.0.2版本

2、按照下面教程下载

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装CUDAcuDNN的步骤如下: 1. 首先,下载CUDA安装文件。可以从NVIDIA官方网站下载对应版本的CUDA安装文件。使用wget命令下载安装文件,并将其移动到指定位置: ``` wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda ``` [1] 2. 接下来,安装显卡驱动。在安装CUDA之前,需要先卸载所有的驱动,并更新软件列表和安装必要的软件和依赖。查询硬件(显卡)信息,并从NVIDIA官网下载对应的驱动。最后,禁用默认驱动(nouveau是通用的驱动程序)。 [2] 3. 安装cuDNN。首先,从NVIDIA官方网站下载对应版本的cuDNN库文件。然后,将相关的库文件拷贝到CUDA安装目录中: ``` sudo cp include/cudnn* /usr/local/cuda/include/ sudo cp lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn* ``` [3] 完成以上步骤后,您的Ubuntu 20.04系统就安装好了CUDAcuDNN。请确保按照上述步骤进行操作,并根据您的系统和硬件配置选择正确的版本。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值