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

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

部署环境:
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

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/916806.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-16
下一篇 2022-05-16

发表评论

登录后才能评论

评论列表(0条)

保存