ubantu20.04和3070ti显卡安装pytorch环境
1.安装显卡驱动
sudo apt-get update
ubuntu-drivers devices //显示当前可用驱动版本
sudo apt-get install nvidia-driver-515 //选择版本并按照
当前可用驱动版本与对应表安装显卡驱动版本
对应表
我这里选择显卡驱动版本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
要等一会,会弹出一个界面,选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}}
更新配置
source ~/.bashrc
确认一下
$ 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