ubuntu20.04 NVIDIA 3060配置CUDA cuDNN anaconda pytorch过程(20230226)

系统 Ubuntu20.04 显卡 英伟达3060 12G

参考资料

B站视频 搭建舒适的Ubuntu使用环境系列——从零到一搭建深度学习生产环境(Pytorch、Tensorflow)并使用GPU版本

优快云 Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)

一、确定版本信息

进入Pytorch官网,确认要安装的CUDA版本
在这里插入图片描述
确定使用CUDA11.6
那么对应的,显卡驱动应该为510

二、安装显卡驱动

打开 software&Updates—>addtional Drivers
在这里插入图片描述
选择510,并点击apply changes

三、安装CUDA

从NVIDIA官网CUDA下载地址下载CUDA11.6.0,选择runfile
在这里插入图片描述

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

接下来按照Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)进行
在这里插入图片描述
在这里插入图片描述
不过,我的.bashrc添加的是

# cuda安装位置
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

四、安装cuDNN

cuDNN下载,注意和CUDA版本匹配,我下载的是Download cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x

在这里插入图片描述
下载deb安装即可

五、安装anaconda

官网下载
在这里插入图片描述
找到下载后的文件,
在终端输入 sh 文件名 即可,注意不要加sudo,以我为例

sh Anaconda3-2022.10-Linux-x86_64.sh  

然后继续进行,按照提示输入yes等,最后会问你要不要初始化,记得选yes(默认是no)

最后,到.bashrc添加

export PATH=$PATH:/home/yourname/anaconda3/bin

六、安装pytorch

按理说,首先要anaconda新建环境,但是我忘了。
我尝试过很多次通过conda安装,搞了一天都没成功。
所以最后选择在打开anaconda的情况下(终端命令行最前面是base)(此时pytorch是装到了anaconda文件夹里面,可通过 pip show torch确认),使用pip安装,一次成功
进入pytorch官网,选择第一步确定的配置,
在这里插入图片描述
然后把官方给出的命令复制到命令行直接执行即可。

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

最后,测试是否安装成功。
终端进入python,

import torch
torch.cuda.is_available()
torch.__version__

结果如下
在这里插入图片描述
应该是成功了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值