Ubuntu 19.10 安装tensorflow gpu 环境
Ubuntu19.10 自带nvidia支持,致使镜像增大到2.+G.
系统Ubuntu19.10
显卡型号RTX2080Ti
- 在"软件和更新"中找到"附加驱动", 选择推荐的驱动, nvidia-driver-435
- 在终端使用nvidia-smi命令查看gpu状态
- 发现右上角有CUDA10.1. Amazing!
- 并且也发现也可以直接安装nvidia-cuda-toolkit
sudo apt-get install nvidia-cuda-toolkit
- 装完之后去下载cudnn(我一直不知道有什么用,直到安装之后查了一下),下载与之配套的cudnn7.7
cudnn:https://developer.nvidia.com/rdp/cudnn-archive
这里我下载的是cuDNN Runtime Library for Ubuntu18.04 (Deb)
- 安装cudnn
sudo dpkg -i libcudnn7_7.6.4.38-1+cuda10.1_amd64.deb
- 安装tensorflow-gpu 2.0.0
- 测试gpu
>>>from tensorflow.python.client import device_lib
>>>device_lib.list_local_devices()