【技术干货】Windows – Tensorflow2.3.0-GPU入坑指南(配置CUDA和CUDNN)
一.背景
- 虽然从英伟达控制面板可以看到我自己电脑安装了conda,但是我并没有nvcc命令,猜测是之前安装cuda时有问题,于是重新安装一遍并配置环境变量。
- 如果直接conda install tensorflow-gpu或者pip install tensorflow都不同程度上出现了一点问题,也就是无法运行tensflow样例程序。
二.安装
安装CUDA,安装在d:\software\CUDA目录下,并且配置环境变量,将bin及lib\x64配置到环境变量中去。
安装CUDNN,这个需要注册才能下载,下载好之后把这四个文件
全部复制到安装目录中去,我的目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1用conda新建一个python3.6的虚拟环境
conda create -n tensorflow python=3.6
然后激活此环境
conda activate tensorflow
并且下载tensorflow和keras
pip install tensorflow
pip install keras
如果有报错缺失dll文件,则自己在网上找这些dll文件下载好后放入到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin中,最常见的缺失cudart64_110.dll,或者相似名称的.dll文件,这时候就自己找链接下载吧,110.dll文件
链接:https://pan.baidu.com/s/1n_YrhR9ZYVgvhPu1zPu4Ng
提取码:3bzh测试代码:
import tensorflow as tf a = tf.constant([1.0, 2.0], name='a') #构建一个二维数组常量 b = tf.constant([2.0, 3.0], name='b') print(a)
结果:
tf.Tensor([1. 2.], shape=(2,), dtype=float32)
欢迎在评论区中进行批评指正,转载请注明来源,如涉及侵权,请联系作者删除。