新手常用Tensorflow配环境代码(整理中)

win10 GTX1060 1070 测试能正常使用的组合
TF-gpu 2.3.1 Keras 2.4.0 cuda 10.1 cudnn 7.6.4.38

下载一次性替换源
pip install tensorflow-gpu==2.1 -i http://pypi.douban.com/simple

查看N卡工作情况
nvidia-smi -l #动态刷新 使用前配路径C:\Program Files\NVIDIA Corporation/NVSMI
查看cuda版本
nvcc -v
查看cuDNN版本
在cuda安装路径下的 Include 文件夹 cudnn.h 中

在编译器中查看GPU
tf.test.gpu_device_name
tf.test.is_gpu_available()

Anaconda 查看环境
conda info --envs
创建新的环境
conda create --name yourENVname python =3.8.0
激活/反激活
conda activate yourENVname
conda deactivate
conda remove yourENVname

关闭所有除erro外的提示
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
设置显存
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.9

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值