Ubuntu 20.04 安装 tensorflow-gpu 1.15.2

不得不说现在部署tf开发环境比以前容易太多了。

安装cuda key

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub

添加库源

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

更新一下

sudo apt-get update

查询cuda版本,成功的话如下图所是。

apt-cache madison cuda

安装cuda

sudo apt-get install cuda=10.0.130-1

安装cudnn,这个文件是在nvdia的网站下载的,注意不要下载错了。

sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb

安装Anaconda3

bash Anaconda3-2020.02-Linux-x86_64.sh

重启电脑,此时如果运行nvidia-smi查看驱动信息,CUDA可能会显示10.2,但实际上安装的还是10.0,无需担心。

安装 tf1.15.2

pip install tensorflow_gpu-1.15.2-cp37-cp37m-manylinux2010_x86_64.whl

结束后输入python,运行以下tf,输出true即可

import tensorflow as tf

print(tf.test.is_gpu_available())

此时建议从ubuntu的软件更新里删除nvidia的安装源,防止不小心更新了CUDA的版本,这样tf 1.15就不能使用了。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容