tensorflow-gpu==2.0.0的安装

****一、tensorflow-gpu==2.0.0的安装
我的版本我cudnn7.6.5 cudn10.0 python 3.6.5(感觉官网的有些匹配不一定对,因为这里cudnn需要为7.6.5)
安装tensorflow的版本号链接
https://tensorflow.google.cn/install/source_windows

Your GPU Compute Capability
您的 GPU 计算能力

https://developer.nvidia.com/cuda-gpus

CUDA下载:https://developer.nvidia.com/cuda-downloads

1、# 查看GPU此时的使用情况
nvidia-smi
注意:下面的CUDA=11.0表示能够支持的最高版本,不是说已经下载好了
(https://www.bilibili.com/video/BV1Vb4y1r78v?from=search&seid=3256408179249773663&spm_id_from=333.337.0.0)
在这里插入图片描述

2、第一种是通过tf.device()函数来指定训练时所要使用的GPU:
tf.device(’/gpu:2’)
3. 通过CUDA_VISIBLE_DEVICES来指定.同样使用第2块GPU来训练模型,我们可以在我们的python代码中加入:
import os
os.environ[‘CUDA_VISIBLE_DEVICES’]=‘2’
4、测试是否成功(注意看tensoflow装在哪个环境,base里面话就在base里面执行下面的命令)
import tensorflow as tf
tf.test.is_gpu_available()

或者:
import tensorflow as tf
import os

os.environ[‘TF_CPP_MIN_LOG_LEVEL’] = ‘2’ # 不显示等级2以下的提示信息

print(‘GPU’, tf.test.is_gpu_available())

a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)
在这里插入图片描述

注意:如果是第一问题,说明的是你的路径没有写道环境变量里面去
https://www.bilibili.com/video/BV1CE411f7wS?from=search&seid=11993867533199687860&spm_id_from=333.337.0.0(v1和v2区别以及解决方法)
在这里插入图片描述

驱动的安装
https://www.nvidia.com/download/index.aspx?lang=en-us
CUDA包的安装路径
https://developer.nvidia.com/cuda-toolkit-archive建议下大的,因为包全一点
默认的CUDA的安装路径(电脑上)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
cmd中输入nvcc -V测试

Cudnn包的安装路径(cudnn神经网络加速包)要注册账号,自己的(784789675@qq.com)
https://developer.nvidia.com/cudnn
把cudnn里面的文件复制到cudn里面去(覆盖)

环境变量添加:系统变量
电脑-属性-高级:
在这里插入图片描述

在这里插入图片描述

pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple(如果有错误,后面这个源也可以去掉)
在这里插入图片描述

Cudnn更详细的步骤:
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-windows
在这里插入图片描述

Conda 的一些常见命令的使用:
Conda list :显示此时环境下安装的包
在这里插入图片描述

conda deactivate tensorflow-gpu 退出当前的环境。
在这里插入图片描述

创建一个新的环境ChipSeq
conda create -n ChipSeq

conda create --prefix="D:\\my_python\\envs\\my_py_env"  python=3.6.3

其中"D:\my_python\envs"是路径名,“my_py_env” 是环境名.
删除创建的环境ChipSeq
conda remove -n ChipSeq --all

需要的cudnn 和 cudn在我传的资源中

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值