如何用anaconda安装cuda和cudnn
时间: 2024-06-01 19:03:11 浏览: 285
安装CUDA和cuDNN可以通过Anaconda进行简化。安装步骤如下:
1.在 https://developer.nvidia.com/cuda-toolkit-archive 下载适用于您的计算机的CUDA版本。请务必选择正确的版本,否则会导致安装失败。
2.在 https://developer.nvidia.com/rdp/cudnn-download 下载cuDNN。也要记住选择适合您的CUDA版本的cuDNN版本。
3.在Anaconda prompt中,输入以下命令创建一个新的虚拟环境:
conda create -n myenv python =3.8
("myenv"是新环境的名称,您也可以选择其他名称)
4.进入新创建的环境:
conda activate myenv
5.安装CUDA和cuDNN:
conda install cudatoolkit =<CUDA版本号>
conda install cudnn =<cuDNN版本号>
例如,如果您下载的是CUDA 11.3,cuDNN v8.2.0,则输入以下命令:
conda install cudatoolkit =11.3
conda install cudnn =8.2.0
6.等待安装完成后,输入以下命令验证CUDA和cuDNN是否正确安装:
nvcc -V # 输出CUDA版本信息
python # 进入Python环境
import torch # 导入torch
torch.cuda.is_available() # 输出True即代表cuDNN成功安装
希望这能帮助回答您的问题! 如果您还有其他问题,请随时问我。
阅读全文