我采用的是Ubuntu18.04，

所以在這里選擇合適的版本：
https://developer.nvidia.com/

選擇完成后給出了命令

下載安裝

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu1804-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1804-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo apt-key add /var/cuda-repo-ubuntu1804-11-2-local/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda

重啟電腦后，用nvidia-smi檢驗(yàn)是否成功

也可以參考tensorflow網(wǎng)站給的類似方法：
https://www.tensorflow.org/install/gpu

后續(xù)的cudnn，參照了這里：https://gist.github.com/mcvarer/30041141c8fe70ea5fe13f839330bc5a

總結(jié)

以上是生活随笔為你收集整理的Tensorflow2.6更新cuda11.2的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。