亲身经历记录:
1.安装anaconda最新版。地址https://www.anaconda.com/distribution/
2.运行anaconda导航,创建新的虚拟环境,起个名字,比如tensorflow-gpu. 注意选择版本,建议安装3.6Python,对应的是tensorflwo-gpu 1.14.0
3.运行anaconda prompt. 输入activate tensorflow-gpu 进入该虚拟环境。输入pip --user --upgrade install tensorflow-gpu。安装。这一步看人品我第一次安装的时候网速较慢,后来晚上又试了一次,超快的速度
4.按照如下安装
- NVIDIA® GPU 驱动程序 - CUDA 10.0 需要 410.x 或更高版本。
- CUDA® 工具包 - TensorFlow 支持 CUDA 10.0(TensorFlow 1.13.0 及更高版本)
- CUDA 工具包附带的 CUPTI。
- cuDNN SDK(7.4.1 及更高版本)
- (可选)TensorRT 5.0,可缩短在某些模型上进行推断的延迟并提高吞吐量。
5.运行anaconda prompt. 输入activate tensorflow-gpu 进入该虚拟环境。输入Python。输入import tensorflow as tf 以及其他测试指令即可知道是否已安装。
import tensorflow as tf
print(tf.reduce_sum(tf.random.normal([1000, 1000])))
默认安装的应该的是你电脑能支持的最高版本。我的电脑如上操作后,安装的是tensorflow-gpu 1.14.0 对应的是Python3.6.9. 如果要指定版本,请查看https://tensorflow.google.cn/install