windows 10 + tensorflow-gpu 环境搭建

 Install Anaconda Python

下载安装Anaconda， https://www.anaconda.com/download/

 安装完成后，在后面的步骤，在Anaconda Prompt中执行

注意：要用管理员权限运行Anaconda Prompt才能完成安装、更新等操作

Update Anaconda

conda update conda
conda update anaconda
conda update python
conda update --all

安装tensorflow-gpu

conda create --name tf-gpu   # Create a Python "virtual environment" for TensorFlow using conda
source activate tf-gpu       # 注意运行此命令后，命令行开头的提示变为(tf-gpu) user@computer:~$，表示tf-gpu环境处于激活状态
# 后面的命令，都在tf-gpu环境下执行，我保留了命令行的提示，以示区别
(tf-gpu) C:\Users\xxx> conda install tensorflow-gpu -y # install TensorFlow with GPU acceleration and all of the dependencies.

 测试安装的tensorflow-gpu

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

不出意外，运行结果会打印

'Hello, TensorFlow!'

 

修改Anaconda Prompt默认env

1. 右键Anaconda Prompt
2. 修改目标target中的activate.bat文件
3. 文件中:END前添加一行： @CALL "%~dp0..\Library\bin\conda.bat" activate tf-gpu

 

修改VSCode中默认环境为tf-gpu

1. 在Settings中搜索，pythonPath
2. 修改 "python.pythonPath"为 "C:\\ProgramData\\Anaconda3\\envs\\tf-gpu\\python.exe"
3. 具体路径视具体情况而定

 

转载于:https://www.cnblogs.com/xbit/p/10228747.html