pytorch 2.0.1 版本 [gpu] 安装

2023-11-15

安装Anaconda

1.下载官网
在这里插入图片描述
我们下载anaconda的目的主要是为了下一步的python环境管理,所以不考虑anaconda与已有的python的版本是否匹配的问题。
在这里插入图片描述
下方点击之后会出现
在这里插入图片描述
可以根据自己的需要下载安装包。
本文以Windows版本的安装包为主。
2.安装
一路next,顺着它的安装步骤来就行。
3.打开
anaconda有两个操作界面:
一个是图形可视化操作界面:Anaconda Navigator
一个是命令行式操作界面:Anaconda Prompt
点开win,找到Anaconda Prompt:
在这里插入图片描述
打开后看见命令行文件路径前有(base),说明安装成功
在这里插入图片描述

虚拟python坏境管理

深度学习中很多项目,由于其依赖包的各种各样,也导致了会出现多个项目使用的python版本不一致的情况。
所以在这里我们就需要建立不同的环境,在不同的需要的时候去使用。类似于电脑上的虚拟机用于运行不同的OS系统。
(1) 打开Anaconda Prompt
(2) 创建一个python版本为3.9的且命名为"env_name"的环境

conda create -n env_name python==3.9

(3)可能会提示让你下载一些包,根据提示决定是否下载就行。

激活/进入 环境()

conda activate env_name

关闭/退出 当前环境()

conda deactivate 

销毁一个 环境() 及其下的所有包

conda remove --name env_name --all

查看有哪些 环境()

conda env list

Pytorch安装

在这里插入图片描述
注意,在安装pytorch前最好安装你想下载的pytorch版本对应的cuda版本,以及cudnn。
可参考文章

1.进入官网下载
可以看到下图
pytorch 2.0.1 仅支持CUDA 11.7 和 CUDA 11.8
可以打开计算机的cmd:
在这里插入图片描述
查看显卡状态。
右上角的CUDA Version:为显卡支持的最高的CUDA版本。
大家可以根据这个去找适合自己的pytorch 版本。
在这里插入图片描述

简单解释:

  • stable:稳定版 Preview:预测版。(一般选择稳定版)
  • Your OS:机器的操作系统
  • 包:conda 和 pip 都可以(不过在anaconda中由于conda下的包比pip下的包优先级更高,故当下载某个包时,一般都是先用conda,不行再用pip)
  • Language:使用的编程语言,也就是c版的pytorch库,和 python版的pytorch库。
  • Computer Platform:CUDA版本可以使用gpu,机器无gpu或不打算用cpu可以直接选择CPU版。
  • 最后一行就是官网根据你的搭配,给出的下载命令,只要在Anaconda Promote的对应环境下输入这段命令即可

注意: -c pytorch -c nvidia 表示前面的包下载的源地址。
考虑到conda的服务器在国外,下载会很慢,
我们 删去 “-c” pytorch ,并添加国内镜像通道,使得 包 pytorch,torchvision,torchaudio从国内镜像网站下载。
(1)conda 清华源
先看官方文档
在这里插入图片描述
根据提示以记事本形式打开用户目录下的.condarc文件
在这里插入图片描述
将下列内容复制到打开的.condarc文件中。

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

注意:
以下述命令为例。
在这里插入图片描述
该命令需要从 pytorch,nvidia 的网址获取包。
我们在custom_channels(候选通道)中找到合适的通道,并添加到channels下;

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

(2)下载pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.7  -c nvidia

下载这里,如果出问题可以在评论区留言

检验是否安装成功

在安装pytorch的命令行里,键入:

python

在打开的python交互界面输入以下代码:

import torch
x = torch.rand(5, 3)
print(x)

正常输出应该类似于

tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])

此外,要检查 GPU 驱动程序和 CUDA 是否已启用并由 PyTorch 访问,运行以下命令以返回是否启用了 CUDA 驱动程序:

import torch
torch.cuda.is_available()

若输出:True 则安装成功

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

pytorch 2.0.1 版本 [gpu] 安装 的相关文章

随机推荐

  • mybatis批量更新

  • 创新工场笔试题----有1分,2分,5分,10分四种硬币,每种硬币数量无限,给定n分钱,求有多少种组合可以组合成n分钱?

    题目 有1分 2分 5分 10分四种硬币 每种硬币数量无限 给定n分钱 求有多少种组合可以组合成n分钱 代码如下 void Combination int a int index int n vector
  • 解决UnicodeEncodeError: 'ascii' codec can't encode characters in position问题

    原文地址 http blog sina com cn s blog 64a3795a01018vyp html 解决方法1 在开头加上 import sys reload sys sys setdefaultencoding utf 8 解
  • Jetpack学习之LiveData

    LiveData是一个可被观察的数据容器类 它是一个数据的容器 将数据包装起来 使数据成为被观察者 当该数据发生变化时 观察者能够获得通知 和ViewModel的关系 ViewModel用于存放页面所需的各种数据 以及数据相关的业务逻辑 因
  • sc 不是内部或外部命令

    安装nodejs 或yarn后 发现控制台命令会提示 sc 不是内部或外部命令 问题原因 nodejs或yarn安装后 会把系统默认环境变量Path的值给替换了 解决方案 将系统环境变量Path的值后面加上 SystemRoot syste
  • java superclass_java Class的 getSuperclass与getGenericSuperclass区别

    Class的getInterfaces与getGenericInterface区别 http www cnblogs com maokun p 6773076 html 一 getSuperclass 返回直接继承的父类 由于编译擦除 没有
  • 【好题】第九届“图灵杯”NEUQ-ACM程序设计竞赛个人赛 F-第二大数 思维

    题 求各个范围的第二大数 举个例子 4 9 6 4 5 6 4 4 6 5 6 第二大数分别是 6 4 4 6 5 6 我们的做法就是按照题目顺序 i从1开始 j从i 1开始 每次记录当前最大数和第二大数 对于每次新出现的数a j 如果它大
  • 二维动态规划>>01背包问题与普遍应用

    0 内容梗概 在二维动态规划中 01背包问题是动态规划中的经典问题 本文首先学习 总结01背包问题的思路 方法与实现 之后 01背包问题与其说是问题 更可以是一种解题思路 或者说套路 如果遇到别的题目时 能够清楚地判断出它是一个01背包的类
  • 【Metashape精品教程2】创建工程

    Metashape精品教程2 创建工程 文章目录 Metashape精品教程2 创建工程 前言 一 导入照片 二 导入pos 三 设置相机参数 没有可不设置 让软件反算相机参数 四 导入控制点 五 保存工程 前言 本章内容主要讲解工程的创建
  • statsmodels中的summary解读(OLS)

    Dep varible y 输出y变量的名称 Model OLS 使用的参数确定的模型OLS Method Least Squares 使用最小二乘法确定参数 Date Sat 10 Aug 2019 日期 Time 18 10 04 时间
  • 什么是Spring Cloud Config

    Spring Cloud Config 是一个解决分布式系统的配置管理方案 它包含 Client和 Server 两个部分 Server 提供配置文件的存储 以接口的形式将配置文件的内容提供出去 Client 通过接口获取数据 并依据此数据
  • 艺术+DeFi,看MEME如何玩转NFT

    内容概述 MEME代币已经成功登陆加密货币交易所Poloniex 就在成功上市之后 MEME代币价格一度飙升至1800美元高位 MEME协议允许人们赚取 菠萝积分 并将其兑换为NFT代币 9月22日 加密货币交易所Poloniex在其DeF
  • 机器学习阶段总结一

    最近一直在看机器学习相关内容 主要是看的斯坦福的Andrew Ng教授的公开课视频 可以点这里 看了前四讲 最大的感觉就是这简直就是概率论 线代 用到的数学知识着实不少 不过也挺有趣的 继续学下去 初步总结一下 主要是关于监督学习 回归分类
  • Redis学习:Redis的持久化

    Redis的持久化方式有两种 rdb和aof rdb理解为快照 save规则出发会的得到一个快照 aof是追加的形式 可以理解为binlog那种 RDB 在进行 RDB 的时候 redis 的主线程是不会做 io 操作的 主线程会 fork
  • 独门秘籍奉上!听说这是CCF遥感地块分割比赛冠军的“获胜法宝”!

    飞桨开发者说 黎昆昌 CCF BDCI遥感影像地块分割赛道冠军团队 CCF BDCI 2020 综合特等奖团队队长 中国科学院深圳先进技术研究院20级硕士 CCF大数据与计算智能大赛 CCF BCI 由中国计算机学会于2013年创办 作为全
  • Nginx配置不缓存html

    Nginx配置不缓存html 1 配置nginx不缓存html 1 1 修改配置文件 1 2 验证 2 Vue项目修改根目录index html 1 配置nginx不缓存html 1 1 修改配置文件 location expires 1h
  • 有特别有创意的网站设计案例

    有人说 UI 设计师集艺术性与科学性于一身 不仅需要对工具的使用熟练 更需要对美术艺术有一定的基础了解 如果想要成为优秀的 UI 设计师是一个需要磨砺的过程 需要不断的学习和积累 多看多练多感受 其中对于优质的设计案例的收集和练习是重要的
  • 清晰、幽默、轻松地深入理解YOLOv5网络结构和一些细节(查阅无数资料文献总结)

    最近的一篇关于YOLOv5检测小目标博客的点击量很高 没想到YOLOv5还是很有影响力的 既然这样 今天本人就本着幽默 清晰 轻松的风格带大家深入了解一下YOLOv5那倾倒众生的网络结构 和它较之其他算法的改进之处 还是一句话 希望我的不经
  • JavaScript的一种单元测试框架:mocha

    mocha 如果你听说过 测试驱动开发 TDD Test Driven Development 单元测试就不陌生 单元测试是用来对一个模块 一个函数或者一个类来进行正确性检验的测试工作 比如对函数abs 我们可以编写出以下几个测试用例 输入
  • pytorch 2.0.1 版本 [gpu] 安装

    安装Anaconda 1 下载官网 我们下载anaconda的目的主要是为了下一步的python环境管理 所以不考虑anaconda与已有的python的版本是否匹配的问题 下方点击之后会出现 可以根据自己的需要下载安装包 本文以Windo