1. 安装显卡驱动
bash
ubuntu-drivers deices
显示所有适配显卡的驱动型号,recommended为推荐安装
- 安装
bash
sudo apt install nvidia-driver-440
- 重启
bash
sudo reboot
- 验证
bash
nvidia-smi
2. 安装cuda
在 CUDA Toolkit 的下载页面选择系统版本和安装方式,下载并运行 runfile。
bash
sudo sh ./cuda_11.7_xxxxxxx.run
显示如果已安装驱动 ,在进行安装的时候取消勾选
根据提示需要修改环境变量
先下载vim
bash
sudo apt install vim
bash
vim ~/.bashrc
在文件末尾添加(这里的/usr/local/cuda-11.3替换成你的cuda安装路径,一般都在/usr/local路径下)
bash
export CUDA_HOME=/usr/local/cuda-11.7
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64
export PATH=$PATH:/usr/local/cuda-11.7/bin
然后更新bashrc
bash
source ~/.bashrc
尝试
bash
nvcc -V
输出对应版本号即成功安装
3. 安装cudnn,nccl
下载对应deb安装包后,使用
bash
sudo dpkg -i xxxxxx.deb
先local,后其他补充包,最后dev
根据提示安装cuda GPG
- 卸载方法
bash
sudo dpkg -r libcudnn8-samples
sudo apt remove package_name
3. 安装anaconda
官网https://www.anaconda.com/products/distribution
linux版本下载完是.sh文件
bash
chmod +x Anaconda3-x.x.x-Linux-x86_64.sh
./Anaconda3-x.x.x-Linux-x86_64.sh
配置环境变量
bash
vim ~/.bashrc
最后一行添加
bash
export PATH=/home/hid/anaconda3/bin:$PATH