一、前言
正如题目所言,最近笔者需要配置一台机器上的深度学习环境,即CUDA工具包和与之对应的cuDNN库,具体步骤如下。
因为我已经装过了,为了教程演示,首先卸载CUDA工具包,利用其自带的卸载程序:
检查本机是否有CUDA工具包,输入nvcc -V
:
二、安装CUDA
-
在命令行输入
nvidia-smi
查看显卡驱动版本也就是最高支持的CUDA工具包版本。
例如,本机可安装11.2及以下的CUDA工具包:
-
在nVidia官网选择对应版本的CUDA工具包并选择你的机器配置,我们就选择11.2.0版本下载,
-
在终端执行如下命令:
wget https://developer.downl