如何在 windows 终端中使用 conda 配置 CUDA 环境
尤记得我此前配置 CUDA 环境都是在各种官网下载各种安装包,然后版本稍微不对就要全部重装。没有对比就没有伤害,这里的方法是我觉得相对更加便捷有效的方法。最近刚从 Linux 重回 Windows 的怀抱,希望将 powershell 也熟悉起来作为一个终端工具,因此以下步骤几乎都是在 Windows powershell 下进行。
确认 CUDA 版本限制
powershell 中先后执行以下命令:
nvidia-smi
在输出的右上角查看 CUDA Version,这是所能允许的 CUDA 最高版本。
安装 anaconda
anaconda 可以用于管理 python 虚拟环境,对于创建独立的 python 开发环境非常重要。先搜索可供下载的 anaconda。
winget search anaconda
复制找到的 ID 号,一般 Anaconda.Anaconda3 是我们想要的。
winget install Anaconda.Anaconda3
conda 环境创建与 CUDA 配置
等待安装完成后,打开 anaconda prompt。
conda create -n CUDA11.8
conda activate CUDA11.8
conda search cudatoolkit
然后选择其中的一个版本,我选的是 11.8,与 pytorch 上的版本对应。
conda install cudatoolkit==11.8.0
接下来安装 cudnn,也是类似的步骤,但是版本要对应。
conda search cudnn
conda install cudnn==8.9.2.26
其中版本对应是很重要的,如果对不上会导致无法使用。接下来安装 pytorch,到其官网,选择好平台,我选的是 Windows conda,之后复制给出的命令到 anaconda prompt 中就可以了。
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
最后就是检验是否成功创建 CUDA 环境,执行以下的命令即可。
python -c "import torch;print(torch.cuda.is_available())"
如果输出为 True
,那么就代表 CUDA 环境配置成功了。恭喜!