我刚刚将系统升级到 Ubuntu 19.04,发现 Cuda 10.0 未安装。tensorflow 仍未针对 Cuda 10.1 构建,如果可以的话,我将 Nvidia 驱动程序从 418 降级到 410,那就太好了。我正在使用 Geforce GTX 1060。
答案1
答案2
请运行这个:
sudo apt-get install cuda-10-0
如果你之前安装过 cuda,
sudo apt-get purge --auto-remove cuda-10-0
并重新安装。
安装后,从存档中下载 cudnn:https://developer.nvidia.com/rdp/cudnn-archive,并解压。之后请执行以下操作:
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/
sudo chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*