conda 安装cudnn
通过 Conda 安装 cuDNN
确保你有 NVIDIA GPU 和 CUDA Toolkit:首先,确保你的系统上安装了 NVIDIA GPU 和 CUDA Toolkit。你可以通过运行以下命令来检查 CUDA 是否已安装:
nvcc --version
如果没有安装 CUDA,你需要先从 NVIDIA CUDA Toolkit 下载并安装。
添加 Conda 通道:为了通过 Conda 安装 cuDNN,你需要添加 NVIDIA 的 Conda 通道。打开你的终端或命令提示符,然后输入:
conda config --add channels conda-forge
conda config --add channels nvidia
搜索并安装 cuDNN:使用以下命令搜索可用的 cuDNN 版本,并安装适合你 CUDA 版本的 cuDNN。例如,如果你的 CUDA 版本是 11.2,你可以搜索并安装对应的 cuDNN 版本:
conda search cudnn -c conda-forge -c nvidia
conda install cudnn -c conda-forge -c nvidia