首页 > 其他分享 >08-Pytorch GPU版详细安装过程

08-Pytorch GPU版详细安装过程

时间:2024-06-15 21:30:53浏览次数:14  
标签:版本 08 torch pytorch Pytorch conda https GPU 安装

1.0 安装Anaconda

官网:https://www.anaconda.com/

安装包下载地址:https://www.anaconda.com/download#downloads

  • 安装步骤

    1710300839518

    1710300867303

    1710300902590

    1710301151779

    image-20240613141354419

     #激活base 虚拟环境
     conda activate base  
     #换源
     pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
     #测试
     pip config list
     pip install requests
    
  • conda基本命令

  #环境本质上就是靠环境变量来实现的
  conda info --env #查看已有环境  
  conda create --name <环境名称> #创建环境
  conda create --name <环境名称> python=<Python版本号>
  
  conda activate <环境名称> #激活环境
  conda deactivate #停用环境包名称
   
  conda install/remove <包名称> #很少使用

练习:创建python3.8.1的环境 并激活

conda create --name lang python=3.8.1
conda info --env
conda activate lang

1.1检查电脑上的显卡型号

image-20240614144924103

注意:只有带NVIDIA的英伟达显卡的电脑才能安装GPU版本的Pytorch,否则只能安装CPU版本的Pytorch.

1.2 CPU版本Pytorch的安装

conda create -n torch-cpu python=3.8.1  #创建虚拟环境
conda activate torch-cpu  #激活虚拟环境
pip install torch torchaudio torchvision  #安装对应包
#验证
python
  import torch
  torch.version.__version__  #'2.3.1+cpu'  有类似信息cpu 版pytorch就安装好了

1.3 GPU版本Pytorch的安装

1.3.1 安装最新版本显卡驱动

https://www.nvidia.cn/geforce/drivers/

image-20240614151055582

下载完直接安装就行。

1.3.2 下载安装CUDA

然后是查看自己电脑驱动的版本,win+r,输入cmd,再输入命令: nvidia-smi

image-20240614151336349

我的CUDA driver的版本就是12.5, 安装CUDA Runtime的版本要小于等于12.5

https://developer.nvidia.com/cuda-toolkit-archive

  • 选择CUDA版本

image-20240614153531135

  • 选择平台

image-20240614153644411

  • 安装

    image-20240614155048855

image-20240614155443970

image-20240614155859550

检查安装是否正常:

image-20240614160223341

配置CUDA涉及的环境变量:

image-20240614160504081

1.3.2 下载安装cuDNN

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10

image-20240614161212710

注意:下载时需要在英伟达官网注册用户。

下载之后,解压缩,将压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下,直接覆盖安装。(其实就是替换)

image-20240614163019284

1.3.3 安装pytorch

https://pytorch.org/

image-20240614163209035

conda create -n torch-gpu python=3.8.1
conda activate torch-gpu
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

如果下载比较慢,可以考虑手工下载:

https://download.pytorch.org/whl/torch_stable.html

image-20240614165953115

可以在https://github.com/pytorch/vision#installation,查询对应版本的torchvision,

image-20240614170410156

然后在https://download.pytorch.org/whl/torch_stable.html

手工下载对应版本的torchvision

image-20240614170913745

可以在https://pytorch.org/get-started/previous-versions/ 查询对应版本的torchaudio

然后在https://download.pytorch.org/whl/torch_stable.html下载

image-20240614171135748

然后通过 pip install 文件名 方式安装下载的这三个文件。

1.3.4 验证

python
import torch
torch.cuda.is_available()
torch.__version__

image-20240614171423711

标签:版本,08,torch,pytorch,Pytorch,conda,https,GPU,安装
From: https://blog.csdn.net/u012046122/article/details/139708702

相关文章

  • P6781 [Ynoi2008] rupq
    P6781[Ynoi2008]rupq线段树上维护这种括号序列,如果信息可差分是好做的,但现在只能合并。先说如何合并信息。max是简单的。至于nand,不需要考虑结合律,只要维护一个bool[32][2]表示当某一位的第一个操作数是0/1时,经过它们的传递、运算的结果是什么。见于P2114[NOI2014]......
  • COMP20008 - Elements of Data Processing
    COMP20008- Elements of Data Processing, Semester 1,2024Assignment2–Whoelse likesthis book?1. OverviewIn thisproject, you willundertake an analysis of a collection of datasets containing detailed informationaboutbooksandt......
  • pytorch学习:安装Anaconda
    安装Anaconda网址Anaconda|TheOperatingSystemforAI然后去邮箱点下载链接(但是为最新版本)建议从下面开始:下载网址推荐历史版本:Anaconda|TheOperatingSystemforAI(更加稳定)不要太新的版本,自己需求选择系统安装鼠标右键管理员身份运行选好自己的路径,......
  • 【NLP-08】NLP的语料库整理
    史上最全开源中文NLP数据集:包括10大类、142条数据源,总有你钟意的那一款,后续将会对数据集进行动态更新;本项目中的NLP数据集囊括了NER、QA、情感分析、文本分类、文本分配、文本摘要、机器翻译、知识图谱、语料库以及阅读理解等10大类共142个数据集。具体而言,对于每......
  • 计算机组成原理:408考研|王道|学习笔记(持续更新~)
    系列目录目录系列目录第一章计算机系统概述1.1计算机的发展(不考)1.2计算机硬件的基本组成1.2.1计算机硬件组成1.2.2各个硬件的工作原理......
  • 第十周:使用PyTorch实现车牌识别
    ......
  • m2_day08 [File]
    课程内容:特等优先级方法特等优先级方法staticlistRoots():列出当前计算机的所有根目录,返回File[]  *:File[]rts=File.listRoots();​list():列出一个目录当中所有的[文件名字],返回String[]   *:Filedir=newFile(".");Stri......
  • 如何快速安装pytorch gpu版本,避免用官方命令下载导致速度过慢
    准备前提:已经安装好CUDA,如果没有安装好CUDA可以查看博主另一篇教程完整安装NVIDIACUDA流程-CSDN博客。为什么要发布此教程:因为在博主安装pytorchgpu版本,耗时了大量时间发现大多数网上教程都是先获取pytorch官方下载命令,进入cmd后去掉-cpytorch等,博主发现不管是conda下载还......
  • 【408精华知识】史上最全的数据结构代码题总结!三万字长文!!!
    关于数据结构代码题,可以说是让很多同学感到头疼了,书上的代码太繁琐、网上的总结不全面让大家对代码题感到云里雾里,那么这篇文章可能会给大家带来一点启发,因为我自己也是深受代码题的折磨,所以一直想写一篇有关它的总结,希望能够做到全面、简洁,让大家用最快的速度记住代码书......
  • 5.08
    <?xmlversion="1.0"encoding="utf-8"?><ScrollViewxmlns:android="http://schemas.android.com/apk/res/android"xmlns:app="http://schemas.android.com/apk/res-auto"xmlns:tools="http://schemas.androi......