当前位置: 首页 > 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/a/288491.html

相关文章:

  • 大数据新视界 -- 大数据大厂之 Impala 存储格式转换:从原理到实践,开启大数据性能优化星际之旅(下)(20/30)
  • PostgreSQL分区表:基础语法与运维实践
  • 利用阿里云下载 WebRTC 源码
  • 九州未来再度入选2024边缘计算TOP100
  • 阿里巴巴通义灵码推出Lingma SWE-GPT:开源模型的性能新标杆
  • MySQL_第13章_视图
  • 【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扫描