（一）深度学习专题-环境安装

一、查询GPU的CUDA版本信息

  • 查询指令：``nvidia-smi`
GPU的CUDA版本信息

二、选用tensorflow-gpu版本

tensorflow版本信息

三、安装CUDA Toolkit

CUDA 安装路径

四、安装cuDNN8.1版本

  • 下载网址：（cuDNN8.1）

  • cuDNN可以看作是CUDA的一个补丁，只要将对应文件拷贝进CUDA的对应文件夹((bin、include、lib))即可。

cuDNN下载页面
CUDA补充文件
CUDA补充文件夹所在位置

五、效果测试

  • 测试指令nucc -V或nvcc --version
运行nvcc -V指令
运行.\bandwidthTest.exe指令
运行.\deviceQuery.exe指令

完成以上步骤，即可完成深度学习的运行环境搭建。接下来便是Anaconda和Tensorflow软件的安装。网上资料比较齐全，此处不再展开描述。

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容