欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

较为详细完整的Tensorflow-gpu环境安装教程

程序员文章站 2024-03-15 10:34:35
...

需求分析

在使用GPU来运行深度学习的项目时,首先需要安装 Tensorflow-gpu 环境。

安装步骤

(1)下载 Anaconda,用以配置 python 环境;

(2)下载 CUDA 和 cuDNN,并安装1
安装过程在这里不再赘述。需要注意的是,CUDA 安装后自动配置环境变量,不需要再进行配置,但cuDNN 需要2

(3)在 Anaconda 的 Anaconda Prompt 窗口里面,进行tensorflow-gpu 的安装,安装命令如下:

conda create --name tensorflow-gpu python=3.6 

【注】可以根据需要自主选择 python 的版本。

安装成功后如下:

Preparing transaction: done
Verifying transaction: done
Executing transaction: done
#
# To activate this environment, use
#
#     $ conda activate tensorflow
#
# To deactivate an active environment, use
#
#     $ conda deactivate

【附】非 gpu 版本 tensorflow 的安装命令:

conda create --name tensorflow python=3.6 

Tensorflow-gpu的**及使用方法

根据安装成功后的提示,在 Anaconda Prompt 里面**tensorflow-gpu 的语句为:

conda activate tensorflow

或者直接输入如下语句命令:

activate tensorflow

以下是在 Anaconda Prompttensorflow-gpu环境下安装库文件的演示:

(tensorflow-gpu) C:\Windows\system32>conda deactivate

(base) C:\Windows\system32>conda activate tensorflow

(tensorflow) C:\Windows\system32>conda deactivate

(base) C:\Windows\system32>activate tensorflow-gpu

(tensorflow-gpu) C:\Windows\system32>pip install matplotlib
Collecting matplotlib
  Downloading matplotlib-3.2.2-cp37-cp37m-win_amd64.whl (9.2 MB)
     |████████████████████████████████| 9.2 MB 437 kB/s

PyCharm的编辑器选择和配置

文章第二小节中中tensorflow-gpu的默认安装目录如下:
较为详细完整的Tensorflow-gpu环境安装教程

所以在 PyCharm 在选择使用 gpu 版本的 tensorflow 运行程序时,要选择该目录下的python.exe程序,如下图所示:
较为详细完整的Tensorflow-gpu环境安装教程

这篇文章写出来,是为了防止后人再走弯路,同时也是为了日后再需要时的查阅方便。
写到这里,差不多本文就要结束了。如果有问题可以在下方留言区留言交流。如果我的这篇文章帮助到了你,那我也会感到很高兴,一个人能走多远,在于与谁同行


  1. Windows环境Tensorflow-cpu/gpu安装教程 ↩︎

  2. Win10安装CUDA10和cuDNN ↩︎