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

一、安装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
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
你可以按照以下步骤来安装condatorchcudacudnn: 1. 首先,安装Anaconda或Miniconda,这是一个用于管理Python环境和软件包的工具。你可以从Anaconda官方网站(https://www.anaconda.com)下载适合你操作系统的版本,并按照官方文档的说明进行安装。 2. 安装CUDA(Compute Unified Device Architecture),这是一个用于利用NVIDIA GPU进行并行计算的平台。你可以从NVIDIA官方网站(https://developer.nvidia.com/cuda-downloads)下载适合你操作系统和GPU的CUDA版本,并按照官方文档的说明进行安装。 3. 安装cuDNNCUDA Deep Neural Network library),这是NVIDIA提供的针对深度神经网络的加速库。你需要先注册一个NVIDIA开发者账号,然后从NVIDIA开发者网站(https://developer.nvidia.com/cudnn下载适合你CUDA版本的cuDNN,并按照官方文档的说明进行安装。 4. 创建一个新的conda环境,并安装PyTorch。你可以使用以下命令创建一个名为"myenv"的新环境: ``` conda create --name myenv ``` 接下来,激活这个环境: ``` conda activate myenv ``` 然后,安装PyTorch和相应的CUDA版本,比如PyTorch 1.9.0: ``` conda install pytorch==1.9.0 torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch ``` 记得将`<your_cuda_version>`替换为你安装CUDA版本,比如`10.2`。 现在,你已经成功安装condatorchcudacudnn,并创建了一个新的conda环境,其中包含了PyTorch和相关依赖。你可以在这个环境中进行深度学习任务了。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

sskay_

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

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

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

打赏作者

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

抵扣说明:

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

余额充值