ubantu20.04和3070ti显卡安装pytorch环境_3070ti支持cuda版本

ubantu20.04和3070ti显卡安装pytorch环境

1.安装显卡驱动

sudo apt-get update
ubuntu-drivers devices                    //显示当前可用驱动版本
sudo apt-get install nvidia-driver-515    //选择版本并按照

1
2
3
4

当前可用驱动版本与对应表安装显卡驱动版本
对应表
我这里选择显卡驱动版本515，cuda为11.6.主要考虑到pytorch安装有与cuda11.6对应的版本

2.安装cuda
如果下载的是cuda的deb安装文件，肯定是要自动安装nvidia-driver的，
为了避免这个事情，不用deb了，改用run文件安装。

$ sudo sh cuda_11.1.6_455.32.00_linux.run

1
2

要等一会，会弹出一个界面，选continue, 然后有个协议，输入"accept“，
之后会弹出如下界面，把第一项的driver去掉，然后选"install"

安装好之后可以看到如下提示

需要配置路径，
在~/.bashrc的最后加入如下两行

export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

1
2
3

更新配置

source ~/.bashrc

确认一下
1
2
3

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.6, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

1
2
3
4
5
6
7

参考
添加链接描述
3.安装cudnn和安装anaconda和pytorch
参考
添加链接描述