三步完成tensorflow2 gpu版本安装配置

最新推荐文章于 2023-04-10 23:54:43 发布

海里的鱼2022

最新推荐文章于 2023-04-10 23:54:43 发布

阅读量481

点赞数 1

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

本文链接：https://blog.csdn.net/pocean2012/article/details/110878684

版权

本文介绍了如何通过conda环境管理器，分三步创建指定Python 3.7版本的虚拟环境，并安装Tensorflow GPU版，最后通过Python验证了GPU的成功识别和使用。

摘要由CSDN通过智能技术生成

经过探索，还是conda安装比较方便，可以安装miniconda，见另一篇文章。

1. 新建虚拟环境，要指定python3.7，之前3.8，3.9都走了弯路

conda create -n py37tf python=3.7

conda activae py37tf

2. 安装tensorflow

conda install tensorflow-gpu

3. 验证

Python 3.7.9 (default, Aug 31 2020, 12:42:55)
[GCC 7.3.0] :: Anaconda, Inc. on linux
>>> import tensorflow
>>> tensorflow.test.is_gpu_available()
WARNING:tensorflow:From <stdin>:1: is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.config.list_physical_devices('GPU')` instead.
2020-12-08 18:00:20.073746: I tensorflow/core/platform/cpu_feature_guard.cc:143] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2020-12-08 18:00:20.102014: I tens