通过conda安装cudatoolikit和cudnn

swrdzwj

已于 2024-08-30 16:38:56 修改

阅读量3k

点赞数 6

文章标签： conda

于 2024-02-20 20:06:37 首次发布

本文链接：https://blog.csdn.net/qq_44886301/article/details/136197752

版权

本文详细介绍了如何通过conda包管理器在Linux系统上安装CUDAToolkit11.3和cuDNN8.2.1，包括cudatoolkit-dev的安装教程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

通过conda安装cudatoolikit和cudnn

查找cudatoolkit

conda search cudatoolkit
conda search cudatoolkit-dev
conda search cudnn

安装cudatoolkit

conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

安装cudnn

conda install cudnn=8.2.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

安装cudatoolkit-dev

conda install cudatoolkit-dev=11.3 -c conda-forge

解决环境中找不到CUDA_HOME的问题

##测试
python
import torch
print(torch.version) # 查看torch版本号
print(torch.version.cuda) # 编译当前版本的torch使用的cuda版本号
print(torch.cuda.is_available()) # 查看当前cuda是否可用于当前版本的Torch，如果输出True，则表示可用