安装CUDA、CUDNN、pytorch

2.1 安装CUDA

2.1.1找到CUDA的版本号

2.1.1.1 屏幕的左下角,找到NVIDIA控制面板的图标,如下图所示:

图2.1.1.1  NVIDIA控制面板图标

2.1.1.2 打开NVIDIA控制面板,找到系统信息,如下图所示:

图2.1.1.2 系统信息

2.1.1.3 在组件里,查看CUDA的版本号,如下图所示:

图2.1.1.3  查看CUDA版本号

2.1.2 安装CUDA

如上图所示,CUDA的版本号是12.2.148,到nvidia的官方网站https://developer.nvidia.com/cuda-toolkit-archive下载对应的版本,如下图所示:

图2.1.2.1 找到CUDA版本

点击链接,进入,按下图所示选择并下载

图2.1.2.1 下载CUDA

下载后,安装即可。

2.2 安装cuDNN

2.2.1 找到cuDNN的版本号

   打开网站https://developer.nvidia.com/rdp/cudnn-archive,找到CUDA的版本号是12.2.148的cuDNN的版本号是V8.9.3(月年都是July 2023),如下图所示:

图2.2.1.1 cuDNN版本号

点击后,选择Windows的版本,如下图所示:

图2.2.1.2  cuDNN Windows版本

点击后,会提示登录,没有账号需要注册,如下图所示:

图2.2.1.3  NVIDIA登录

2.2.2 安装cuDNN

1)下载后,解压,如下图所示:

​图2.2.2.1  cuDNN解压后文件夹

2)找到CUDA的安装路径,一般在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2,如下图所示:

图2.2.2.2  CUDA安装路径

        将cuDNN解压后的bin、include、lib三个文件夹,复制粘贴到CUDA的安装文件里。 

2.3 安装PyTorch

   1)打开PyTorch的官网PyTorch,并按图选择, 如下图所示:

图2.3.1  PyTorch网站

   2)在应用中找到并打开Anaconda Prompt,输入conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia如下图所示:

图2.3.2 运行安装pytorch指令

安装成功,如下图所示:

图2.3.3 pytorch安装成功

  1. 如果安装失败,选择CPU,如下图所示:

图2.3.4  pytorch安装选择CPU

4)打开Anaconda Prompt,输入conda install pytorch torchvision torchaudio cpuonly -c pytorch

  • 7
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Anaconda是一个基于Python的开发环境,包含常用的Python第三方库和工具。安装Anaconda可以方便地创建和管理Python虚拟环境,使得项目之间的依赖不会互相影响。安装Anaconda可以到官网下载相应版本的安装包,根据安装向导完成安装即可。 CUDA是NVIDIA推出的一个并行计算平台和编程模型,用于加速GPU计算。安装CUDA需要先确认自己的GPU型号和对应的CUDA版本,然后到NVIDIA官网下载相应版本的CUDA安装包,安装过程中需要注意选择安装目录和添加环境变量,完成安装后需要验证CUDA是否安装成功。 cuDNN是NVIDIA开发的深度神经网络(DNN)加速库,提供了高效的DNN前向和反向算法的实现。安装cuDNN需要先下载对应版本的cuDNN安装包,并将解压后的文件复制到CUDA安装目录下,然后添加cuDNN的路径到环境变量中。 PyTorch是一个基于Python的开源机器学习框架,支持动态计算图和自动求导。安装PyTorch可以通过pip或conda命令进行安装,建议使用conda install命令,可以解决依赖关系问题。在安装PyTorch时需要注意选择对应的Python版本和CUDA版本,以及安装适配的cuDNN版本。 总之,安装Anaconda、CUDAcuDNNPyTorch需要仔细阅读各自的安装指南,并根据自己的硬件和软件环境选择适配的版本和配置。同时要注意添加环境变量和验证安装是否成功,避免安装失败或出现兼容性问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

orangapple

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

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

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

打赏作者

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

抵扣说明:

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

余额充值