主要是在conda下搭建GPU版本的TensorFlow,后续会提及Keras的安装
TensorFlow_GPU 1.15的安装
conda安装
anaconda官网下载安装即可。
创建conda环境
创建一个新的环境,主要是为了万一出错只删除这个环境就行,不影响你原来的Python环境,如果你足够自信的话,可以不用创建
conda create -n "你的环境名" python = python版本号
![](http://101.201.142.67:12/wp-content/uploads/2020/07/1.png)
![](http://101.201.142.67:12/wp-content/uploads/2020/07/2.png)
conda activate + 你的环境名
成功进入环境后,如下图所示
![](http://101.201.142.67:12/wp-content/uploads/2020/07/3.png)
pip安装1.15版本的TensorFlow-GPU
使用阿里的源进行安装
pip install tensorflow-gpu==1.15 -i http://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com
由于网络的原因,可能会失败,多试几次就可。
以下还有一些其他的国内源