Tensorflow-gpu安装，ubuntu20.04

西柚西柚xiu

已于 2023-02-20 20:24:06 修改

阅读量258

点赞数

分类专栏： tensorflow 文章标签： tensorflow python 深度学习

于 2023-02-20 20:23:07 首次发布

本文链接：https://blog.csdn.net/tanking66/article/details/129130993

版权

tensorflow 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

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()