1. 创建 Conda 环境
首先,创建一个新的 Conda 环境(可选):
conda create -n myenv python=3.8
conda activate myenv
2. 安装 CUDA 11.0
使用以下命令安装 CUDA 11.0:
conda install cudatoolkit=11.0
3. 安装 cuDNN
接下来,安装与 CUDA 11.0 兼容的 cuDNN。你可以通过以下命令安装 cuDNN:
conda install cudnn=8.0.5=cuda11.0_0
请确保使用与 CUDA 11.0 版本匹配的 cuDNN 版本。你可以在 Anaconda Repository 上查看可用的版本:
conda search cudnn
4. 验证安装
安装完成后,可以通过以下方式验证 CUDA 和 cuDNN 是否安装成功:
python -c "import torch; print(torch.cuda.is_available())"
如果你使用的是 TensorFlow,可以使用以下命令检查:
python -c "import tensorflow as tf; print(tf.test.is_gpu_available())"
5. 配置环境变量(可选)
如果需要,还可以手动配置环境变量,但在 Conda 中通常不需要这样做,因为 Conda 会自动管理这些设置。
6. 注意事项
- 确保你的系统上已安装 NVIDIA 驱动,支持 CUDA 11.0。
- 在选择 cuDNN 版本时,请参考 NVIDIA 的 cuDNN 版本文档 以确保与所安装的 CUDA 版本兼容。
按照上述步骤,你应该能够在 Conda 环境中成功安装 CUDA 11.0 和 cuDNN。如果遇到问题,请随时告诉我!
如果在使用 conda install
时遇到安装失败的情况,增加镜像源可以提高下载速度和成功率。以下是如何在 Conda 中添加镜像源的步骤: