ubunu中配置torch环境4060显卡

一、第一种安装方法
1、查看系统版本和显卡型号

cat /etc/issue # 查看系统版本
lspci | grep -i nvidia # 查看电脑的显卡型号
1
2

2、查看系统是否安装显卡驱动 nvidia-driver及显卡支持的驱动型号

nvidia-smi # 查看显卡驱动情况（此处未安装，应无信息显示）
ubuntu-drivers devices # 查看显卡支持的驱动型号
1
2

3、安装推荐版本的显卡驱动

sudo ubuntu-drivers autoinstall # 安装默认显卡驱动　　
1

4、检查驱动是否安装成功后直接安装cuda

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run
1
2

5、下载cudnn并解压，然后使用如下命令将include和lib64中的文件移动到cuda文件夹当中

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda-11.3/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.3/lib64/libcudnn*
1
2
3
4