使用CUDA或者tensorflow-gpu时,一般需要电脑安装对应的NVIDIA驱动。Ubuntu的PPA安装方式如下:
-
添加PPA源
sudo add-apt-repository ppa:graphics-drivers/ppa
-
更新并安装对应的版本
sudo apt-get update
sudo apt-get install nvidia-driver-410 #此处要根据上面查询到的版本适当更改
sudo apt-get install mesa-common-dev (optional)
sudo apt-get install freeglut3-dev (optional)
-
测试安装是否成功
nvidia-smi 或者 nvidia-settings