Tensorflow-gpu安装,ubuntu20.04

前置条件

1、安装conda
2、降级CUDA版本为cuda10.1

创建环境

创建一个python版本为3.7版本的新环境

conda create -n tf-g-2 python==3.7

创建成功

安装TensorFlow-gpu

conda install tensorflow-gpu

安装成功

测试是否能调用

(tf-g-2) liuyq@liuyq-MS-7D42:~$ python
Python 3.7.0 (default, Oct  9 2018, 10:31:47) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
>>> tensorflow.test.is_gpu_available()
True
>>> exit()

输出为True证明调用成功

更多推荐