手把手教你快速安装gpu版本的pytorch(详细图文教程)

 更新时间:2024年09月27日 09:37:38   作者:北舟zhou  
在Windows 10上安装PyTorch时,通常默认安装的是CPU版本,且下载速度较慢,本文提供了一个详细的安装指南,包括如何检查CUDA版本、选择合适的PyTorch、torchvision和torchaudio版本,并通过pip而非conda进行安装,文中通过图文介绍的非常详细,需要的朋友可以参考下

最近在win10装torch,发现没有Linux那么便捷,用conda和pip装默认版本都是cpu,而且下载速度很慢。网上的解决方法有好几种,但是初学者可能面临的问题是:我的系统该装哪个版本的?cuda、torch、torchvision以及 torchaudio装哪个版本?以及装了cpu版本的显示torch.cuda.is_available()  False 该怎么办?

那么,方法如下(亲测成功好用)

第一步:创建环境 (-n +环境名(torch) + python版本(3.10))

conda create -n torch python=3.8

第二步:了解你电脑的cuda 版本,在terminal中输入:nvidia-smi 即可,如下图所示,版本为12.3

第三步:去pytorch官网看你的conda适配的torch、torchvision以及 torchaudio版本,这个链接:

Previous PyTorch Versions | PyTorch

比你的CUDA低版本的都能安装,看好对应的版本就可以了,不用官网的命令安装(会很慢)

第四步:开始安装,鉴于使用conda的各种问题,我们这里不使用conda安装,使用pip安装,根据刚才看好的版本,在创建的torch环境下输入命令:

#进入torch环境
conda activate torch
#用镜像源快速安装对应版本
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

第五步:验证torch安装是否成功,import torch成功说明安装成功,如果torch.cuda.is_available()返回False 时,说明装了cpu版本的。不要慌,往下看。

第五步:查看torch环境下所有库:

conda list

可以看到,pytorch是cpu版本的

第六步:手动去清华源下载对应的gpu版pytorch,和上面图上看到的cpu的版本保持一致即可

Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

比如pytorch版本是1.10.1(需要对应),环境创建时python版本是3.8 (需要对应),cuda 12.3(比它小的都可以),那么下面几个可以下载,选一个下载,最好装到一个英文路径下。

第七步:离线安装刚刚下载的文件

#  cd 到下载的文件目录,例如安装目录是E:\Download
cd E:\Download
#离线安装
conda install --offline pytorch-1.10.0-py3.6_cuda10.2_cudnn7.6.5_0.tar.bz2

最后一步: conda list, 查看pytorch 是否已经被替换。可以看到替换成GPU版本了

再输入 torch.cuda.is_available() 验证,返回True .大功告成!

总结

到此这篇关于快速安装gpu版本的pytorch的文章就介绍到这了,更多相关快速安装gpu版本pytorch内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • python 多维切片之冒号和三个点的用法介绍

    python 多维切片之冒号和三个点的用法介绍

    下面小编就为大家分享一篇python 多维切片之冒号和三个点的用法介绍,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • 在树莓派2或树莓派B+上安装Python和OpenCV的教程

    在树莓派2或树莓派B+上安装Python和OpenCV的教程

    这篇文章主要介绍了在树莓派2或树莓派B+上安装Python和OpenCV的教程,主要基于GTK库,并以Python2.7和OpenCV 2.4.X版本的安装作为示例,需要的朋友可以参考下
    2015-03-03
  • Python获取与处理文件路径/目录路径实例代码

    Python获取与处理文件路径/目录路径实例代码

    我们在用python进行数据处理时往往需要将文件中的数据取出来做一些处理,下面这篇文章主要给大家介绍了关于Python获取与处理文件路径/目录路径的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2023-04-04
  • 玩转python爬虫之cookie使用方法

    玩转python爬虫之cookie使用方法

    Cookie用于服务器实现会话,用户登录及相关功能时进行状态管理,这篇文章主要介绍了使用python处理cookie的方法,感兴趣的小伙伴们可以参考一下
    2016-02-02
  • Python实现LSTM学习的三维轨迹

    Python实现LSTM学习的三维轨迹

    这篇文章主要为大家详细介绍了如何使用LSTM来学习和预测三维轨迹,并提供详细的Python实现示例,感兴趣的小伙伴可以跟随小编一起学习一下
    2024-12-12
  • Python轻松管理与操作文件的技巧分享

    Python轻松管理与操作文件的技巧分享

    在日常开发中,我们经常会遇到需要对文件进行操作的场景,如读写文件、文件夹操作等。本文将为大家介绍一些 Python 中处理文件的实用技巧,让你的工作更高效
    2023-05-05
  • Django管理员账号和密码忘记的完美解决方法

    Django管理员账号和密码忘记的完美解决方法

    这篇文章主要给大家介绍了关于Django管理员账号和密码忘记的完美解决方法,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面来一起看看吧
    2018-12-12
  • python selenium模拟点击问题解决方案

    python selenium模拟点击问题解决方案

    这篇文章主要介绍了python selenium模拟点击问题,涉及到安装谷歌浏览器和浏览器驱动的相关知识介绍,本文结合实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-05-05
  • pythotn条件分支与循环详解

    pythotn条件分支与循环详解

    这篇文章主要介绍了Python条件分支和循环用法,结合实例形式较为详细的分析了Python逻辑运算操作符,条件分支语句,循环语句等功能与基本用法,需要的朋友可以参考下
    2021-08-08
  • 搭建python django虚拟环境完整步骤详解

    搭建python django虚拟环境完整步骤详解

    这篇文章主要介绍了搭建python django虚拟环境完整步骤详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-07-07

最新评论