降级 CUDA 而不更改 NVIDIA 驱动程序版本

降级 CUDA 而不更改 NVIDIA 驱动程序版本

我正在努力降级我当前的 CUDA 版本。我正在使用带有 NVIDIA GeForce RTX 3070 GPU、460 驱动程序和 CUDA 11.2 的 Ubuntu 20.04 LTS。我正在使用 tensorflow 1.13.1 作为机器学习软件包的一部分,由于某种原因,该软件无法正常工作。我怀疑这是因为 CUDA,因为我使用带有 NVIDIA TITAN V GPU、450 驱动程序和 CUDA 11.0 的相同软件,并且该软件运行良好。

我首先尝试将 NVIDIA 驱动程序降级到 450,因为这会自动安装 CUDA 11.0。但是,RTX 3070 GPU 似乎仅支持 460 驱动程序,因此降级驱动程序不是一个选择。

接下来,我尝试仅降级 CUDA,而不触及驱动程序。首先,我尝试删除当前的 CUDA 安装:

sudo apt-get --purge remove "*cublas*" "cuda*"

然后从NVIDIA 档案使用该.deb (local)文件(按照网站上的安装说明):

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.0.3/local_installers/cuda-repo-ubuntu2004-11-0-local_11.0.3-450.51.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-0-local_11.0.3-450.51.06-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

然而,这似乎总是会自动将驱动程序恢复到 450,从而导致依赖冲突。基于网站,CUDA 11 应该支持 >=450 驱动程序,那么是否可以在不更改驱动程序的情况下将 CUDA 11.2 降级到 11.0?

非常感谢!

相关内容