PyTorch 环境配置

目录

  • [一、安装 CUDA](#一、安装 CUDA)
  • [二、安装 PyTorch](#二、安装 PyTorch)
    • [1. 创建虚拟环境](#1. 创建虚拟环境)
    • [2. 安装 PyTorch](#2. 安装 PyTorch)
  • [三、在 PyCharm 上创建一个 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 :
  • 配置环境变量:系统属性 → 环境变量 → 系统变量 → 双击 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 是否安装成功:确保进入已激活的 PyTorch 虚拟环境中,依次输入以下代码并执行
bash 复制代码
python  # 进入 Python 环境,若需要退出 Python 环境,请输入 exit()
import torch 
print(torch.__version__)  # 输出安装的 PyTorch 版本号
print(torch.cuda.is_available())  # 成功检测到 PyTorch 并可以使用 GPU

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

  • 文件 → 新建项目
  • 验证是否成功
相关推荐
CoookeCola10 小时前
开源图像与视频过曝检测工具:HSV色彩空间分析与时序平滑处理技术详解
人工智能·深度学习·算法·目标检测·计算机视觉·开源·音视频
董厂长10 小时前
综述:deepSeek-OCR,paddle-OCR,VLM
人工智能·计算机视觉
gfdgd xi11 小时前
deepin 终端,但是版本是 deepin 15 的
linux·python·架构·ssh·bash·shell·deepin
禁默11 小时前
基于金仓KFS工具,破解多数据并存,浙人医改造实战医疗信创
数据库·人工智能·金仓数据库
云卓SKYDROID11 小时前
无人机动力学模块技术要点与难点
人工智能·无人机·材质·高科技·云卓科技
王六岁11 小时前
🐍 前端开发 0 基础学 Python 入门指南:条件语句篇
前端·python
java1234_小锋11 小时前
PyTorch2 Python深度学习 - 初识PyTorch2,实现一个简单的线性神经网络
开发语言·python·深度学习·pytorch2
胡萝卜3.011 小时前
C++面向对象继承全面解析:不能被继承的类、多继承、菱形虚拟继承与设计模式实践
开发语言·c++·人工智能·stl·继承·菱形继承·组合vs继承
Danileaf_Guo11 小时前
29瓦功耗运行140亿参数模型!Mac mini M4的AI能效革命
人工智能
小草cys11 小时前
华为910B服务器(搭载昇腾Ascend 910B AI 芯片的AI服务器查看服务器终端信息
服务器·人工智能·华为·昇腾·910b