安装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

相关推荐
三无推导11 小时前
ComfyUI 安装部署教程:Windows 下快速搭建可视化 AI 绘图工作流,零基础也能跑通
人工智能·pytorch·windows·stable diffusion·aigc·ai绘画·持续部署
独隅12 小时前
PyTorch自动微分模块:从原理到实战一
人工智能·pytorch·python
不羁的木木19 小时前
HarmonyOS文件基础服务(Core File Kit)实战演练03-文件增删改查与目录操作
pytorch·华为·harmonyos
盼小辉丶20 小时前
PyTorch深度学习实战(55)——在Android上部署PyTorch模型
android·pytorch·python·模型部署
zhendianluli1 天前
PyTorch 复杂模型转 ONNX 踩坑纪实:从 diff 到 nan_to_num 的三关突破
人工智能·pytorch·python
weixin_468466852 天前
PyTorch 与 TensorFlow 实战选型与应用场景指南
人工智能·pytorch·深度学习·算法·机器学习·tensorflow·深度学习框架
独隅2 天前
PyTorch 新手从零搭建深度学习环境实战指南
人工智能·pytorch·深度学习
keineahnung23452 天前
在 Google Colab 中安裝 PyTorch 2.2.0
人工智能·pytorch·python·深度学习
AI算法沐枫2 天前
机器学习经典小项目1:鸢尾花分类
人工智能·pytorch·深度学习·神经网络·机器学习·分类·数据挖掘
weixin_468466852 天前
PyTorch 深度学习框架核心能力与实战评测
人工智能·pytorch·深度学习·神经网络·计算机视觉·动态图·模型训练