基于Anaconda3下安装TensorFlow-gpu的方法和遇到的坑

最新推荐文章于 2024-06-04 17:27:42 发布

reisaru

最新推荐文章于 2024-06-04 17:27:42 发布

阅读量8k

点赞数 1

分类专栏： TensorFlow 文章标签：机器学习 TensorFlow

本文链接：https://blog.csdn.net/qq_41337100/article/details/86710551

版权

前提条件：已经装好了anaconda3，CUDA和CUDNN，只针对TensorFlow本身的坑。

首先创建python环境，给这个python环境取名叫TensorFlow。 conda create -n tensorflow python=3.6（后文我的环境叫mytensorflow是因为是第二次安装）
再这之前建议配置一下从镜像源下载，不然即使是科学上网也顶不住。
直接打开anaconda的命令行输入即可。
```
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/  
conda config --set show_channel_urls yes  
```
同样一打开anaconda就可以执行命令anaconda search -t conda tensorflow会显示出目前可供选择的tensorflow版本，按平台选择版本

在这里插入图片描述

执行命令anaconda show conda-forge/tensorflow获取对应版本的安装命令conda-forge/tensorflow为选择的版本名称
因为官方源太慢了，即使我科学上网也很慢，所以用了中科大的镜像源。选择的是我能使用的最新的版本。这里贴出来的表是windows的。

https://blog.csdn.net/oMoDao1/article/details/83241074#commentBox
这里面有很全的版本对应关系。

pip install tensorflow-gpu==1.10.0

import tensorflow as tf

关注