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

部署环境:
RTX3060+win11

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

【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
安装包下载路径:cuda11.5,大家自己下载吧
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
解压后 默认精简安装即可
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
安装cudnn 8.4
https://developer.nvidia.com/rdp/cudnn-download
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
下载完成后,将文件解压,里面会有三个文件夹bin, include, lib, 将这三个文件夹里面的文件分别复制到cuda的安装目录下面,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
在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

【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
pip install tensorflow-gpu==2.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
或者直接阿里云下载后直接安装(传送门
【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2
查看版本:
import tensorflow as tf

print(tf.__version__)

a = tf.test.is_built_with_cuda()
b = tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)

print(a)
print(b)

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

Original: https://blog.csdn.net/weixin_44322778/article/details/124639745
Author: 时间之里
Title: 【深度学习】RTX3060+CUDA 安装 win11+cuda 11.5+tensorflow-gpu 2.6.2

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496514/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球