Conda+CUDA+Torch+cuDNN下载及安装(总结)

本文详细介绍了如何安装Anaconda,包括卸载旧版本、选择合适的版本及配置环境变量。接着,文章指导了CUDA的下载与安装,强调了版本选择的重要性。然后,重点讲解了在Anaconda中创建环境并安装PyTorch,以及验证安装成功的步骤。最后,文章阐述了cuDNN的下载、解压和安装到CUDA目录的过程,并提供了验证安装是否正确的方法。
摘要由CSDN通过智能技术生成

一、安装Anaconda

如果安装过需卸载,直接以管理员身份运行uninstall-anaconda3可彻底删除。

选择版本,网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/(建议别下最新版)

  1. 默认环境安装(不容易出问题)-->just me-->no add PATH environment variable;

  1. 环境变量配置

cmd-->where conda-->where python;

  1. 测试是否安装成功:进入cmd,输入语句:conda --version

二、安装CUDA(针对有GPU,可选择安装)

安装网址:https://developer.nvidia.com/cuda-downloads

1.选择版本,Archive of Previous CUDA Releases

2. CUDA的版本选择,跟Torch版本要求对应,不会出错。

3. 根据自己电脑环境,推荐迅雷下载,然后默认地址安装(不容易出问题)。

三、安装pytorch

  1. 进入anaconda,创建torch的环境,python版本推荐3.8。

  1. Win+R输入cmd,并输入activate base;activate torch激活自己的环境。

  1. torch官网https://pytorch.org/get-started/locally/,选择对应版本的torch。

  1. 复制上面的一段代码,至Command,进入自动安装。

  1. 在Command验证安装是否成功。

输入:

import torch

print(torch.__version__) #显示版本

print("gpu", torch.cuda.is_available()) #False

  1. 至此torch安装成功,并可调用gpu加速。

四、安装cuDNN

  1. https://developer.nvidia.com/cudnn (注册账号)

  1. 进入下载界面:

  1. 根据 cuDNN Archive | NVIDIA Developer ,查看适配的 cuDNN

我的是CUDA v11.7,因此可以选择较新的cuDNN v8.8。

  1. 下载window的压缩包

  1. 解压得到cuDNN的文件夹

  1. 将把三个文件夹拷贝到cuda的安装目录下(如下图为默认路径)

对三个不同文件夹内容分别进行拷贝。

  1. 验证:win+R启动cmd,然后cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe(进到目录后需要直接输“bandwidthTest.exe”和“deviceQuery.exe”),得到下图:

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
您可以按照以下步骤下载安装CUDNN: 1. 登录到NVIDIA开发者网站:https://developer.nvidia.com/。 2. 在顶部菜单中选择 "登录",使用您的NVIDIA帐户登录。如果没有帐户,您可以在网站上注册一个新帐户。 3. 登录后,将鼠标悬停在顶部菜单上的 "下载" 上,然后选择 "所有下载"。 4. 在 "操作系统" 下拉菜单中选择适合您系统的操作系统类型。 5. 在 "类别" 下拉菜单中选择 "加速库"。 6. 滚动页面并找到 "cuDNN",然后点击它。 7. 在 "cuDNN 下载" 页面上,选择适合您系统的 cuDNN 版本,并点击 "下载"。 8. 下载完成后,解压缩下载的文件。 安装CUDNN: 1. 打开终端或命令提示符,进入解压缩的文件夹。 2. 复制 cuDNN 文件到 CUDA 安装目录。例如,在 Linux 上,默认的 CUDA 安装目录是 /usr/local/cuda。您可以使用以下命令复制文件: ``` sudo cp -P include/cudnn*.h /usr/local/cuda/include sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* ``` 3. 添加 cuDNN 到环境变量。编辑您的 .bashrc 文件,并在文件末尾添加以下行: ``` export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 4. 保存并关闭 .bashrc 文件,然后运行以下命令使更改生效: ``` source ~/.bashrc ``` 5. 现在,您已成功安装 cuDNN。您可以在代码中引入 cuDNN 库并使用它。 请注意,安装 cuDNN 需要先安装 CUDA,因为 cuDNNCUDA 的一个加速库。在安装 cuDNN 之前,请确保已正确安装并配置了 CUDA
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

sskay_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值