pytorch的gpu环境安装

windows系统下pytorch的gpu环境安装

安装前置说明:

  1. 要成功安装pytorch的gpu环境,需要cuda版本,python版本和pytorch版本都要相匹配才能顺利使用,cuda版本建议不要安装最新的版本,以免找不到相匹配的pytorch版本
  2. python版本尽量不要安装最新的,以免找不到相匹配的pytorch版本
  3. torch,torchvision,torchaudio三者版本对应关系:参阅

1. 安装cuda和cuDNN

Cuda和cuDNN安装教程

2. 根据已经安装的cuda版本和python版本选择pythorch版本,并下载安装包安装

2.1 查看自己已经安装的cuda版本和python版本

cuda版本为12.4,要下载的版本为cu124

python的版本为3.9,要下载的版本为cp38

2.2 下载安装包

点击如下链接下载安装包:
pytorch安装包下载地址

  1. 下载torch,点击torch -- 查找Ctr F -- 内容cu124-cp39,点击对应的安装包下载
    note:可以复制链接--打开迅雷下载,提高下载速度
  2. 下载torchaudio
  3. 下载torchvision

2.3 本地命令行安装

pip install torch-2.4.0+cu124-cp39-cp39-win_amd64.whl

pip install torchvision-0.19.0+cu124-cp39-cp39-win_amd64.whl

pip install torchaudio-2.4.0+cu124-cp39-cp39-win_amd64.whl

查看是否安装成功:

相关推荐
dhxhsgrx10 分钟前
PYTHON训练营DAY27
开发语言·python
I"ll carry you13 分钟前
【2025.5.12】视觉语言模型 (更好、更快、更强)
人工智能·语言模型·自然语言处理
双翌视觉27 分钟前
机器视觉光源选型解析:照亮工业检测的“智慧之眼”
人工智能·机器视觉·视觉对位·视觉软件
Echo``35 分钟前
1:OpenCV—图像基础
c++·图像处理·人工智能·opencv·算法·计算机视觉·视觉检测
FL1717131437 分钟前
MATLAB机器人系统工具箱中的loadrobot和importrobot
人工智能·matlab·机器人
☞无能盖世♛逞何英雄☜43 分钟前
Flask框架搭建
后端·python·flask
Q_Q19632884751 小时前
python的家教课程管理系统
开发语言·spring boot·python·django·flask·node.js·php
夏天是冰红茶1 小时前
图像处理:预览并绘制图像细节
图像处理·人工智能·opencv
点云SLAM1 小时前
Python中in和is关键字详解和使用
开发语言·人工智能·python·python学习·in和is关键字·python中for循环
后知后觉1 小时前
深度学习-最简单的Demo-直接运行
人工智能·深度学习