文章目录[隐藏]
有一块显卡,怎么在Ubuntu中安装驱动和cuda那些依赖?
要快速安装。
如果之前安装了,就卸载:
### 卸载
sudo apt remove --purge nvidia-*
卸载依赖
sudo apt autoremove --purge
版本如何对应
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#id3
驱动版本最低的适配版本
不想细看,就使用下面命令行输入:
Nvidia Driver
sudo apt-get nvidia-common
sudo add-apt-repository ppa:graphics-drivers
sudo apt update
ubuntu-drivers devices
sudo apt install nvidia-driver-530
ubuntu-drivers devices:
cuda-toolkit
版本:
https://developer.nvidia.com/cuda-toolkit-archive
直接选最新的:
https://developer.nvidia.com/cuda-downloads
本地安装:
就是下载到本地后安装,本地会有一个deb文件
根据上图选择好后,下面会有一串操作命令,挨个输入就可以:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.1-530.30.02-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.1-530.30.02-1_amd64.deb sudo cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/ sudo apt-get update sudo apt-get -y install cuda
搞完重启: sudo reboot
安装CuDNN
https://developer.nvidia.com/rdp/cudnn-download
下载下来压缩包
点击查看下载哪个版本更合适:
如何查看目前系统和硬件信息?
查看Ubuntu信息
cat /proc/version
显示linux的内核版本和系统是多少位的:X86_64代表系统是64位的:
uname -a
要查看分区情况,那么输入命令:sudo fdisk -l
free -m 命令,这样就可以在命令字符界面查看内存的情况
解压: tar -xvf 文件.tar.gz
然后拷贝:
cd 文件夹名
sudo cp include/* /usr/local/cuda-12.1/include
sudo cp lib/libcudnn* /usr/local/cuda-12.1/lib64
sudo chmod a+r /usr/local/cuda-12.1/include/cudnn*
sudo chmod a+r /usr/local/cuda-12.1/lib64/libcudnn*
最后
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
显示:
配置环境
sudo vim ~/.bashrc
在最后添加:
export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib64
生效重载:
source ~/.bashrc
sudo ldconfig
验证
验证显卡驱动版本
cat /proc/driver/nvidia/version
查看:
nvcc -V
nvidia-smi
其他
如果项目需要的cuda、Torch版本等比较低怎么办?
对每个环境可以单独安装cuda版本,上面安装的是系统的驱动和依赖包
我们可以使用conda或者pip安装。
conda和pip 环境内安装卸载
conda安装:conda install cuda -c nvidia
conda卸载:conda remove cuda
安装版本:conda install cuda -c nvidia/label/cuda-11.3.0
pip安装:
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#pip-wheels
python3 -m pip install --upgrade setuptools pip wheel
python3 -m pip install nvidia-pyindex
|requirements.txt
file: --extra-index-url <https://pypi.org/simple
>
python3 -m pip install nvidia-cuda-runtime-cu12
我把12.1相关的驱动和安装包放在网盘了,需要自取
https://pan.baidu.com/s/1jM-JtLA3vtjB_1zDvUy3GA?pwd=9i4n
最后欢迎关注公众号: