当前位置: 首页 > news >正文

pytorch安装的超级详细教程(没有之一)

一、发展历程 (简单介绍)

(15年)caffe --> (16年)tensorflow1.x --> (17年)keras --> (18年)Tensorflow2.x --> (19年)pytorch。

面向gihub开源项目编程。

向下支持比较好,各个版本之间支持比较好,兼容性强。

版本分为GPU和CPU两种

二、配置环境(cpu)(比较容易)

pip install -U torch -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
pip install -U torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

三、配置环境(GPU)(比较复杂)

1.查看自己的卡适合哪个版本的CUDA

cmd命令如下

nvdia-smi

 建议:优先装11.x,因为该公司在更新的时候,可能对10.x版本就不再进行支持了。

2.安装CUDA

直接在百度搜CUDA(下面这篇博客详细的介绍了如何安装CUDA,亲测有效)

手把手教你安装CUDA(一看就会)_cuda安装_时代&信念的博客-CSDN博客

安装之后,在cmd中输入

nvcc-V

出现下列界面,证明安装成功

2.安装pytorch

1)直接在百度搜pytorch

PyTorch

然后我们在下面的这个网站里面来配置pytorch

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

按照自己的cuda 和python的版本去下载torch、torchvistion

1、2、3、4分别代表的是CUDA的版本、python版本、操作系统的类型、操作系统的位数。

将上面的两个文件安装到anacoda的sciprits下面

在这个'D:\anacoda\Scripts>'路径下pip 上面的包

torchvision-0.11.0+cu113-cp37-cp37m-win_amd64.whl
pip install torch-1.10.2+cu113-cp37-cp37m-win_amd64.whl

 上面的命令要根据自己下载的torch和torchvision进行改变

2)检查是否安装成功:

输入下列命令:

python
print(torch.cuda.is_available())

若出现下面的’TRUE‘,则证明安装成功

至此pytorch配置完成。

http://www.lryc.cn/news/32278.html

相关文章:

  • leetcode两数之和
  • 断点调试(debug)
  • Vuex 状态管理
  • 射频功率放大器在辉光放电特征及风速测量原理中的应用
  • keepalived+nginx 双机热备搭建
  • 零基础如何入门学习Python?
  • 认识CSS之基础选择器
  • JavaScript面试题整理汇总
  • 科普| 什么是云原生?
  • Spring Boot 3.0系列【9】核心特性篇之依赖管理
  • 每日算法题
  • xss靶场绕过
  • 【数据结构】核心数据结构之二叉堆的原理及实现
  • Spring Cloud Alibaba+saas企业架构技术选型+架构全景业务图 + 架构典型部署方案
  • RocketMQ-03
  • 大神教你在 Linux 中查看你的时区
  • Redis持久化策略
  • 显著性检验【t-test、方差分析、ks检验】
  • 访问学者在德国访学生活衣食住行攻略
  • SQL-刷题技巧-删除重复记录
  • 基于JSP的虚拟账号交易平台
  • LeetCode201_201. 数字范围按位与
  • 一款好的风险管理软件可以做什么
  • html2canvas使用文档
  • HTML DOM 改变 CSS
  • 基于EB工具的TC3xx_MCAL配置开发01_WDG模块配置介绍
  • Activty启动到显示的过程[二]
  • ubuntu 18.04.06LST安装R4.0+版本报错及解决过程
  • 数据湖架构Hudi(五)Hudi集成Flink案例详解
  • 【Java学习笔记】9.Java 循环结构 - for, while 及 do...while