手把手教你快速安装gpu版本的pytorch(详细图文教程)
最近在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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
文章主要介绍了在VSCode中使用Jupyter Notebook遇到的问题及其解决过程,问题包括包版本冲突、文件冲突、路径错误和找不到文件,通过逐一排查和安装相关依赖包,最终解决了这些问题,使得Jupyter Notebook可以在VSCode中正常运行2025-02-02
使用jupyter notebook直接打开.md格式的文件
这篇文章主要介绍了使用jupyter notebook直接打开.md格式的文件，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
Python利用wxPython模块打造ChatGPT式打字效果程序
这篇文章主要为大家介绍了如何利用Python和wxPython模块打造一个ChatGPT式打字效果程序，从而增强用户体验或提高应用程序的可读性，感兴趣的可以了解一下2023-05-05
tensorflow基于CNN实战mnist手写识别(小白必看)
这篇文章主要介绍了tensorflow基于CNN实战mnist手写识别(小白必看),文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-07-07
这篇文章主要介绍了Python如何使用argparse模块处理命令行参数,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
这篇文章主要介绍了Python基础详解之列表复制,文中有非常详细的代码示例,对正在学习python基础的小伙伴们有很好的帮助,需要的朋友可以参考下2021-04-04
在软件开发过程中,IPO（Input-Process-Output）分析是一种常用的方法,用于分析软件系统的输入、处理和输出,Python作为一种功能强大的编程语言,提供了许多工具和库来实现IPO分析,本文将介绍如何使用Python进行IPO分析的基本步骤和常用工具,需要的朋友可以参考下2023-12-12
详解Pandas如何高效对比处理DataFrame的两列数据
我们在用 pandas 处理数据的时候，经常会遇到用其中一列数据替换另一列数据的场景。这一类的需求估计很多人都遇到，当然还有其它更复杂的。解决这类需求的办法有很多，这里我们来推荐几个2022-09-09
这篇文章主要介绍了python re模块findall()函数实例解析，具有一定借鉴价值,需要的朋友可以参考下2018-01-01
Python的Flask框架中的Jinja2模板引擎学习教程
这篇文章主要介绍了Python的Flask框架中的Jinja2模板引擎学习教程,Jinja2模板引擎的用法也是Flask的Web开发中的重要知识,需要的朋友可以参考下2016-06-06
最新评论