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