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

查看是否安装成功:

相关推荐
S01d13r几秒前
gunicorn + flask 处理高并发请求
python·flask·gunicorn
杜子不疼.5 分钟前
《Python列表和元组:从入门到花式操作指南》
开发语言·python
荼蘼5 分钟前
机器学习之PCA降维
人工智能·机器学习
pan0c2311 分钟前
数据处理与统计分析 —— numpy入门
python·numpy
东方不败之鸭梨的测试笔记18 分钟前
智能测试用例生成工具设计
人工智能·ai·langchain
max50060027 分钟前
基于桥梁三维模型的无人机检测路径规划系统设计与实现
前端·javascript·python·算法·无人机·easyui
秋氘渔1 小时前
综合案例:Python 函数知识整合 — 学生成绩管理系统
开发语言·python
AI 嗯啦2 小时前
SQL详细语法教程(三)mysql的函数知识
android·开发语言·数据库·python·sql·mysql
databook2 小时前
把数学对象画出来:Manim Mobject类库速查手册
python·数学·动效
失散133 小时前
深度学习——02 PyTorch
人工智能·pytorch·深度学习