部署环境: 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
查看版本:
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)
|