用conda创建tensorflow-gpu

最新推荐文章于 2024-06-21 10:00:00 发布

蓝雨飞扬7

最新推荐文章于 2024-06-21 10:00:00 发布

阅读量644

点赞数 1

文章标签： tensorflow 深度学习 windows

本文链接：https://blog.csdn.net/qq_42921511/article/details/120854805

版权

本文详细指导如何在Windows上为Tensorflow GPU安装环境，包括安装步骤、推荐版本（CUDA 10.0 + cuDNN 7.6 + TensorFlow 1.14.0），并介绍了使用Anaconda创建和激活环境，验证GPU支持的方法，以及设置国内镜像源。

摘要由CSDN通过智能技术生成

在安装Tensorflow-gpu之前，我们需要安装cuda和cudnn等。对于Tensorflow的gpu版本最重要的是配方（CUDA、cuDNN、Tensorflow版本的搭配）。
详情见网址：https://tensorflow.google.cn/install/source_windows最下方
请添加图片描述
使用Anaconda来创建环境（建议python3.6+cuda10+cudnn7.6+tensorflow1.14.0为例）

conda create -n tensorflow1.14 python=3.6
activate tensorflow1.14
conda install cudatoolkit=10.0
conda install cudnn=7.4
conda install tensorflow-gpu==1.14.0

验证

python
import tensorflow as tf
tf.test.is_gpu_available()

pip还源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

然后贴几个国内源：
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

蓝雨飞扬7

关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
用conda创建tensorflow-gpu

在安装Tensorflow-gpu之前，我们需要安装cuda和cudnn等。对于Tensorflow的gpu版本最重要的是配方（CUDA、cuDNN、Tensorflow版本的搭配）。详情见网址：https://tensorflow.google.cn/install/source_windows最下方使用Anaconda来创建环境（建议python3.6+cuda10+cudnn7.6+tensorflow1.14.0为例）conda create -n tensorflow1.14 python
复制链接

扫一扫