RTX3090+Ubuntu安装CUDA问题
RTX3090的算力是sm86,目前测试适合安装470版本的驱动。安装好驱动后,输nvidia-smi
查看信息:
上面写的CUDA Version只是建议安装版本,并不是你已经安装了对应的版本,这里还是需要去NVIDIA官网去找到对应驱动11.3,以命令行安装。
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
注意上面几行要一行一行复制粘贴执行,一下子全粘贴上是不行的。