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

PyTorch 环境配置

目录

  • 一、安装 CUDA
  • 二、安装 PyTorch
    • 1. 创建虚拟环境
    • 2. 安装 PyTorch
  • 三、在 PyCharm 上创建一个 PyTorch 项目


参考文章:
【2025年最新PyTorch环境配置保姆级教程(附安装包)】
【超详细 CUDA 安装与卸载教程(图文教程)】
【PyTorch安装、配置环境(全网最新最全)】

默认已经安装好了 Anaconda 以及 PyCharm ,win + r 输入 cmd ,使用 conda --version 命令查看 Anaconda 的版本;python --version 命令查看 Python 的版本。

一、安装 CUDA

  • 检查显卡型号:计算机管理 → 设备管理器 → 显示适配器

  • win + r 输入 cmd ,使用 nvidia-smi 命令显示显卡驱动版本和 cuda 版本。

  • 下载 CUDA 的网址:【CUDA Toolkit Archive】

  • 安装 CUDA :

  • 配置环境变量:系统属性 → 环境变量 → 系统变量 → 双击 Path ,根据 CUDA 安装的路径新建以下几条(如果没有)。
D:\CUDA
D:\CUDA\v12.8\bin
D:\CUDA\v12.8\libnvvp

  • 验证 CUDA 是否安装成功:打开命令提示符(Win + R,输入 cmd),输入 nvcc --version 命令,如果命令行成功显示了 CUDA 的版本信息,则表明 CUDA 安装和环境变量配置已成功。

二、安装 PyTorch

1. 创建虚拟环境

参考文章:【在 Anaconda 上安装多版本 Python 环境并在 PyCharm 中配置】

  • 打开 Anaconda Prompt ,输入命令 conda create -n [名称] python=[版本] 创建一个新的虚拟环境。

  • 环境创建完成后,输入命令 conda activate [环境名称] 激活该环境。

2. 安装 PyTorch

  • 获取 PyTorch 安装命令:【Get Started】

  • 验证 PyTorch 是否安装成功:确保进入已激活的 PyTorch 虚拟环境中,依次输入以下代码并执行
python  # 进入 Python 环境,若需要退出 Python 环境,请输入 exit()
import torch 
print(torch.__version__)  # 输出安装的 PyTorch 版本号
print(torch.cuda.is_available())  # 成功检测到 PyTorch 并可以使用 GPU

三、在 PyCharm 上创建一个 PyTorch 项目

  • 文件 → 新建项目

  • 验证是否成功

http://www.dtcms.com/a/343234.html

相关文章:

  • Telnet、ftp详解
  • 教育场景下禁用html5播放器拖动进度条的例子
  • python 项目编号 2025821 有关于中英文数据的收集、处理
  • mac的m3芯片通过Homebrew安装git
  • ES_分词
  • 2025-08-21 Python进阶9——__main__与lambda
  • Harbor私有仓库实战配置
  • FLUX-Text模型完全配置指南:从环境搭建到故障排除
  • 用例完备性1:用例模板
  • 数据结构-HashMap
  • Kubernetes“城市规划”指南:告别资源拥堵与预算超支,打造高效云原生都市
  • Typora 快速使用入门:15分钟掌握高效写作
  • 锅炉铸造件三维扫描尺寸及形位公差检测技术方案-中科米堆CASAIM
  • ⸢ 啟 ⸥ ⤳ 为什么要开这个专栏?
  • Ubuntu Server 系统安装 Docker
  • uni-app:实现文本框的自动换行
  • SpringBoot + Vue实现批量导入导出功能的标准方案
  • k8sday13数据存储(1.5/2)
  • 基于Matlab多技术融合的红外图像增强方法研究
  • C++---滑动窗口平滑数据
  • 瑞派亚宠展专访 | 以数智化重塑就医体验,共筑宠物健康新生态
  • 区块链存证操作
  • echarts关系图(Vue3)节点背景图连线设置
  • 2025.7.19卡码刷题-回溯算法-组合
  • IOS购买订阅通知信息解析说明Java
  • 设计模式3-模板方法模式
  • 爬虫基础学习-项目实践:每次请求,跟换不同的user-agent
  • 茶饮业内卷破局,从人力管理入手
  • iOS 手势与控件事件冲突解决清单
  • 一本通1342:【例4-1】最短路径问题