【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2

部署环境:
RTX3060+win11
在这里插入图片描述

在这里插入图片描述
安装包下载路径:cuda11.5,大家自己下载吧
在这里插入图片描述
解压后 默认精简安装即可
在这里插入图片描述
在这里插入图片描述
安装cudnn 8.4
https://developer.nvidia.com/rdp/cudnn-download
在这里插入图片描述
下载完成后,将文件解压,里面会有三个文件夹bin, include, lib, 将这三个文件夹里面的文件分别复制到cuda的安装目录下面,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5
在这里插入图片描述
在path中添加
c:\program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\bin
c:\program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\lib\x64

windows pycharm环境配置:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

在这里插入图片描述
pip install tensorflow-gpu==2.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
或者直接阿里云下载后直接安装(传送门
在这里插入图片描述
查看版本:

import tensorflow as tf

print(tf.__version__)   # 返回 true

a = tf.test.is_built_with_cuda()  # 判断CUDA是否可以用
b = tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)                                  # 判断GPU是否可以用

print(a)   # 返回 true
print(b)   # 返回 true

在这里插入图片描述

  • 4
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 9
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

时间之里

好东西就应该拿出来大家共享

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值