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

【PyTorch】安装,环境搭建

PyTorch是深度学习最常使用的框架之一,使用框架可以极大的提高效率。
安装步骤

  1. 创建环境
    使用anaconda创建环境
    安装conda以及使用可以参考Python 环境搭建——anaconda环境隔离以及在pycharm中使用
conda create -n test  创建
conda env list 查看环境

在指定的环境中安装需要切换到对应的环境目录
conda activate 环境名
2. 官网查看安装方法
PyTorch官网
在这里插入图片描述

定位到网页,根据自己的情况选择对应的版本,复制命令到对应的conda环境
Previous versions of PyTorch可以找到历史版本
3. CUDA版本
如果电脑有独立显卡,可以选择CUDA版本使用显卡加速模型训练。
首先对显卡的驱动进行更新
在这里插入图片描述

查看电脑的显卡版本,确定Compute Platform的CUDA版本,驱动版本需要高于官网的版本
是用cmd 的nvidia-smi命令查看版本,以及一些自己的显卡信息
在这里插入图片描述
4. 确定自己的版本信息,以及自己需要的PyTorch 版本。复制命令进行安装
5. 测试
安装完成导入进行测试

import torch
device = "cuda" if torch.cuda.is_available() else "cpu" # 如果CUDA可用就是cuda,否则是cpu
print(device)  #cuda
print(torch.__version__) # 2.3.0  版本

torch.cuda.is_available() 返回布尔值,cuda是否可用


http://www.kler.cn/news/288491.html

相关文章:

  • 【Netty】实战:基于Http的Web服务器
  • 数据分析及应用:如何分析基于绝对中位差的异常值检测问题?
  • LINUX网络编程:Tcpsocket封装
  • Java GC机制:Minor GC与Full GC的触发条件
  • 假期作业--数据结构
  • uniapp插槽用法
  • vue子组件样式影响父组件
  • 每天一个数据分析题(五百一十六)- 贝叶斯分类算法
  • Axure打造科技感数据可视化大屏原型
  • 网络安全宗旨和目标
  • OpenCV颜色空间转换(1)颜色空间转换函数cvtColor()的使用
  • 【论文阅读】skill code 和 one-shot manipulate
  • C++ 设计模式——职责链模式
  • Go父类调用子类方法(虚函数调用)
  • stm32之I2C通信外设
  • 提升RAG检索回答质量: Shortwave的 4 大优化指南
  • 使用 Milvus Lite、Llama3 和 LlamaIndex 搭建 RAG 应用
  • 住宅IP与机房IP:哪种更适合业务应用?
  • 51单片机-第十节-独立按键及数码管优化
  • shell脚本—————局域网IP扫描
  • 开放式耳机漏音有多大?五大超值爆款推荐!
  • 【数据可视化技术】使用Matplotlib、Seaborn进行数据可视化
  • 9、Django Admin优化查询
  • 【2025考研英语高分写作:写作核心词汇】三、人生哲理类
  • Django中的第一个自动化测试编写
  • VS Code CMake tool 配置
  • Apple LLM: 智能基础语言模型(AFM)
  • Apache Struts2开发模式漏洞解析与修复
  • 温湿度传感器和IOT网关的连接 485接口
  • 嵌入式OTG硬件电路分析