服务器安装cuDNN
参考链接:CUDA官方安装文档
pytorch安装
CUDA 11.2
显卡 3090
ubuntu 18.04
-
安装cuDNN(deb文件)
我没有导入秘钥,update时有报错(应该是网络问题),也可以成功安装 -
测试cuDNN
第3步编译时报错 fatal error: FreeImage.h: No such file or directory
解决办法:执行sudo apt-get install libfreeimage3 libfreeimage-dev
,重新编译后运行./mnistCUDNN
,出现Test passed!至此cuDNN安装完成
在这里能看见cudnn的版本
验证深度学习环境
参考:知乎 深度学习环境搭建
如何才能验证深度学习的环境(我的是cuda11.2+cudnn8.1.1+tensorflow-gpu 2.7)是OK的呢?
#执行一下Python脚本进行验证
import tensorflow as tf
print(tf.test.is_built_with_cuda())
#若输出为true则表示cuda安装成功
print(tf.config.list_physical_devices('GPU')) #这一栏我改过
#若输出为true则表示能够使用gpu