大体思路按照下面的链接完成
Ubuntu18.04安装TensorFlow-gpu2.0_dl_None-CSDN博客_ubuntu安装tensorflow2.0
记录几点我自己的心得体会
1.首先说下我自己的环境
操作系统:Ubuntu18.04
cuda版本:10.1
cudnn版本:7.6.5
pytorch版本:1.4
Tensorflow版本:2.1需要前置安装:pycharm+anaconda3
2.如果在使用conda安装时,速度太慢,则可以考虑用pip的国内源安装
例如:
清华源:https://pypi.tuna./simple
阿里源:/pypi/simple/
豆瓣源:/simple/
注:新版ubuntu要使用https
使用方法:pip install xx -i https://pypi.tuna./simple
3.自己使用的cuda cudnn tensorflow的版本要兼容,需要在tensorflow官网查询
/in...
4.安装过程中要注意ubuntu的权限问题
5.在完成tensorflow-gpu安装后,需要把cudnn中的libcudnn.so.7文件拷贝到
/usr/local/lib
6.在新版本的tensorflow中,用以下代码验证tensorflow-gpu安装成功与否
tf.config.list_physical_devices('GPU')