目录

  • 1.Anaconda安装
  • (1) Anaconda下载安装
  • (2)(可选)Anaconda Navigator启动其他程序
  • 2.安装TensorFlow-GPU版本
  • 3.安装相应的CUDA 和cuDNN 版本


1.Anaconda安装

(1) Anaconda下载安装


官网下载速度还行,一路默认安装,直到这一步骤,一定记得勾选添加anaconda到Path中,不然后面还有要自己手动添加path。

vscode怎么换tensorflow版本_CUDA


后续一切就绪,完成安装后,在cmd中输入

conda -V

以测试Anaconda是否成功安装

vscode怎么换tensorflow版本_CUDA_02

(2)(可选)Anaconda Navigator启动其他程序

打开Anaconda Navigator

vscode怎么换tensorflow版本_CUDA_03


Anaconda Navigator可以用来方便地启动应用、方便的管理conda包、环境和频道,不需要使用命令行的命令。

vscode怎么换tensorflow版本_tensorflow_04


启动在页面中启动VScode可省去配置的麻烦

2.安装TensorFlow-GPU版本

首先在cmd窗口在创建一个虚拟环境

conda create -n tensorflow pip python=3.7  #虚拟环境的名称为tensorflow

遇到y/n时选择y

安装完成后,使用命令

activate tensorflow 进入虚拟环境,我们将在此虚拟环境中安装tensorflow GPU版本

vscode怎么换tensorflow版本_CUDA_05


接下来利用anaconda的pip’工具安装TensorFlow的gpu版本,直接用清华源来下载,不然速度会很慢很慢,这里使用命令

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu

等待安装完成后还需要下载NVIDIA的CUDA 和cuDNN

3.安装相应的CUDA 和cuDNN 版本

首先选择合适的CUDA版本(根据你需要的TENSORFLOW型号,选择需要的CUDA,可在TENSORFLOW官网查询)

 

vscode怎么换tensorflow版本_tensorflow_06


下载得到安装程序,运行后选择默认地址即可(此地址为临时地址,安装结束后会删除,不必担心占用c盘空间,同时注意,若自定义安装路径,则不能与此时的临时路径相同,否则会被删除)