ubuntu-drivers devices
如果显卡驱动版本过低或不兼容,可以使用以下命令安装最新的NVIDIA驱动程序:
sudo apt-get update
sudo ubuntu-drivers autoinstall
安装CUDA:
sudo apt-get install cuda
安装cuDNN:
下载合适版本的cuDNN,并且将文件解压到相应目录下:
tar -xzvf cudnn-10.0-linux-x64-v7.5.0.56.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
pip install tensorflow-gpu==1.15
希望上述步骤可以帮助您解决TensorFlow-GPU在AWS EC2实例上无法正常工作的问题。