安装nvidia驱动
1、查看可用的驱动
ubuntu-drivers devices
2、安装nvidia驱动
sudo apt install nvidia-driver-xxx
3、重启电脑可以看到显卡显示为NVIDIA Corporation
安装cuda
1、查看nvidia驱动安装情况并查看cuda的可安装版本
nvidia-smi
结果显示可以安装cuda12.2以下的版本
2、下载cuda安装程序
(1)进入cuda官网
(2)选择低于12.2的的cuda版本
(3)运行图片中两行指令
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
(4)按一次“向下”键,按enter
(5)输入accept,按enter
(6)按一次enter(不选择Driver),再按几次“向下”键,按enter
(7)安装完成
3、添加cuda相关的环境变量
(1)打开.bashrc配置文件
vim ~/.bashrc
(2)在文件最后写
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export NUMBAPRO_NVVM=/usr/local/cuda/nvvm/lib64/libnvvm.so
export NUMBAPRO_LIBDEVICE=/usr/local/cuda/nvvm/libdevice/
(3)刷新配置
source ~/.bashrc
安装cudnn
1、进入官网
2、选择Local Installer for Linux x86_64(Tar),这一步需要在官网注册一下,跟着引导点就行了
(1)下载好之后解压
tar -xvf cudnn-linux-x86_64-8.9.5.30_cuda11-archive.tar.xz
(2)进入cudnn文件夹,依次输入以下命令
cd cudnn-linux-x86_64-8.9.5.30_cuda11-archive
sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
显示下图表示安装成功了
安装anaconda
1、进入官网
### 2、选择下载64-Bit (x86) Installer (1007.9M)
输入以下命令
chmod +x Anaconda3-2024.06-1-Linux-x86_64.sh
# 前面不要加sudo,否则会安装到root目录下,需要管理员权限才能使用conda
./Anaconda3-5.3.0-Linux-x86_64.sh
# 安装过程中按enter和yes,不要写no,否则conda的环境变量需要自己配置
3、重启shell终端,查看conda版本
conda -V
pip换清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
conda换清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --set show_channel_urls yes