TensorFlow笔记 安装

安装过程很曲折······

首先,Python必须是3.5-3.7版本的:
具体见TensorFlow官网
在这里插入图片描述
而现在的Anaconda的Python是3.8版本的:
在这里插入图片描述
这可能是我之前下载TensorFlow库一直失败的原因之一。(看官方网站/文档很重要!
所以,我在清华大学开源软件镜像站上下载Anaconda的4.2.0版本
4.2.0版本的Anaconda用的Python是3.5.2版本的。

然后,安装CUDA和cuDNN:
具体的安装过程和环境变量的配置略,有很多博客都说得很清楚了。
需要注意的是,CUDA一定要选择10.0版本的,然后cuDNN也一定要选对应于CUDA 10.0的。

最后,安装TensorFlow:
看清自己的电脑能否安装tensorflow-gpu。
打开cmd安装tensorflow-gpu:

pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple

但我遇到了这个错误:
在这里插入图片描述
即:Cannot remove entries from nonexistent file d:\codingtools\anaconda3\lib\site-packages\easy-install.pth

我打开它所说得d:\codingtools\anaconda3\lib\site-packages路径,下面并没有easy-install.pth这个文件,于是创建了一个,然后执行如下指令:

pip uninstall setuptools
pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple

然后就安装成功了:
在这里插入图片描述
打开ipython测试一下:
在这里插入图片描述

测试用的py程序:

import tensorflow as tf
import os

# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

print('GPU:', tf.test.is_gpu_available())

a = tf.constant(2.)
b = tf.constant(3.)

print(a+b)
print(a*b)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值