一、查询GPU的CUDA版本信息
- 查询指令:``nvidia-smi`
二、选用tensorflow-gpu版本
- 根据tensorflow_gpu-2.6.0下载CUDA11.2版本和cuDNN8.1版本。(Tensorflow对应版本)
三、安装CUDA Toolkit
- 下载网址:(CUDA Toolkit 11.2)
四、安装cuDNN8.1版本
下载网址:(cuDNN8.1)
cuDNN可以看作是CUDA的一个补丁,只要将对应文件拷贝进CUDA的对应文件夹((bin、include、lib))即可。
五、效果测试
- 测试指令
nucc -V
或nvcc --version
完成以上步骤,即可完成深度学习的运行环境搭建。接下来便是Anaconda和Tensorflow软件的安装。网上资料比较齐全,此处不再展开描述。