news 2026/8/21 12:30:15

Pytorch安装教程

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Pytorch安装教程

1、创建虚拟环境

1)在cmd中

工作目录:D:\DaiMa\PythonCode\LLMS\LLMs-from-scratch-main

虚拟空间:D:\DaiMa\PythonCode\LLMS\LLMs-from-scratch-main\.venv

激活虚拟空间:进入工作目录,打开cmd,输入 .venv\Scripts\activate.bat

输入where python(where pip)显示使用的python和pip应来源于虚拟空间,用uv包管理器创建环境的话的话,pip不来自虚拟空间,因为uv极简,不带pip

1、安装uv

pip install uv

2、创建虚拟环境

uv venv --python=python3.11

3、激活虚拟环境

.venv\Scripts\activate.bat

4、下载jupyter

pip install jupyter -i https://pypi.mirrors.ustc.edu.cn/simple/

5、打开jupyter

uv run jupyter lab

2)在pycharm中

新建项目New Project,在解释器类型选择项目venv创建虚拟环境

打开Terminal终端,路径前有(.venv)即表示虚拟环境已激活

2、安装pytorch

PyTorch官网,找到和自己cuda版本和驱动版本匹配的pytorch,

历史版本网页:Previous PyTorch Versions

在cmd中输入nvidia-smi可以查看CUDA版本和驱动版本

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

官网下载太慢

通过Ctrl+C终止下载,复制找到的下载地址,用迅雷下载

或者使用镜像文件下载

镜像文件下载地址:https://download.pytorch.org/whl/torch/

下载完成后,在终端输入以下语句,进行本地下载

pip install D:\DaiMa\PythonCode\Learning_Pytorch\torch-2.0.1+cu117-cp311-cp311-win_amd64.whl

如果失败可能是因为,默认从清华源下载,改为使用以下语句即可

pip install D:\DaiMa\PythonCode\Learning_Pytorch\torch-2.0.1+cu117-cp311-cp311-win_amd64.whl -i https://pypi.org/simple

这样就是下载成功了

在控制台输入以下语句,输出True证明GPU版的torch已经正常安装!

import torch torch.cuda.is_available()

3、安装torchvision、torchaudio

在 历史版本网页:Previous PyTorch Versions 中找到与自己CUDA版本相匹配的历史版本,复制该命令,把torch删掉,已经下载过了

pip install torchvision==0.15.2 torchaudio==2.0.2

由于默认在清华源下载,还是报错,改为

pip install torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple

其含义是

  • --index-url https://download.pytorch.org/whl/cu117:告诉 pip 优先从 PyTorch 官方源(CUDA 11.7 版本)查找这几个主包。torchvision==0.15.2torchaudio==2.0.2通常是对应 CUDA 11.7 的版本组合。

  • --extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple:将清华源设置为备用源。如果这些包还有其他依赖(比如numpypillow),pip 会从清华源下载,这样速度会更快

这样就下载成功啦!

4、安装matplotlib pandas

清华源还是没有,改用阿里云下载

pip install matplotlib pandas -i https://mirrors.aliyun.com/pypi/simple/

5、查看

使用这个命令行,检查安装的包

pip list

安装完成!

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/21 12:29:24

PPO强化学习在ESP32平衡机器人上的工程落地指南

这类项目最值得先看的不是理论推导,而是能不能在普通开发板上跑起来、能不能稳定收敛、以及从仿真到实物的坑点在哪里。用PPO训练一个平衡机器人,听起来像是学术研究,但核心价值在于把强化学习从“跑通Demo”推进到“在真实硬件上完成闭环控制…

作者头像 李华
网站建设 2026/8/21 12:28:01

本地AI音频工具实战:从环境部署到批量处理的完整落地指南

这类工具最值得先看的不是功能列表,而是能不能在普通环境里稳定跑起来。我一般会先确认它到底解决的是转写、配音还是字幕生成问题,然后看本地跑起来需要什么条件,单条任务怎么验证,批量处理时又要注意哪些坑。 下面按实际落地顺…

作者头像 李华
网站建设 2026/8/21 12:27:58

AI Agent 模型迁移实战:从 Claude 到 GLM 的架构适配与工程化落地

这类项目最值得关注的不是“从 Anthropic 换到 GLM”这个动作本身,而是背后关于 AI Agent 开发、模型选型、成本控制、稳定性保障和工程化落地 的一整套实战经验。如果你正在用 Claude Opus、GPT-4 这类顶级模型做 Agent 开发,但面临成本高、响应慢、或…

作者头像 李华
网站建设 2026/8/21 12:26:45

生产环境部署Qwen3.8-9B-GGUF:服务化、监控与高可用的完整指南

生产环境部署Qwen3.8-9B-GGUF:服务化、监控与高可用的完整指南 【免费下载链接】Qwen3.8-9B-GGUF 项目地址: https://ai.gitcode.com/hf_mirrors/empero-ai/Qwen3.8-9B-GGUF Qwen3.8-9B-GGUF 是 Empero 团队基于 Qwen3.8-9B 蒸馏模型推出的 GGUF 量化版本&a…

作者头像 李华
网站建设 2026/8/21 12:25:38

Java全栈开发面试实战:从基础到架构的完整方案

1. Java全栈开发面试实战概述 作为一名经历过上百场技术面试的Java全栈开发者,我深刻理解面试准备与实际工作能力之间的鸿沟。市面上大多数面试资料要么停留在八股文层面,要么过于理论化,缺乏真实项目落地的衔接。这篇文章将分享我从基础到架…

作者头像 李华
网站建设 2026/8/21 12:23:53

TensorFlow神经网络二分类实战:从数学建模到金融风控预测

1. 项目概述:当数学建模遇上神经网络二分类 在数学建模竞赛和实际数据分析项目中,二分类问题可以说是“家常便饭”。无论是预测用户是否会点击广告、判断一封邮件是否为垃圾邮件,还是诊断某种疾病的风险,其核心都是将样本划分到两…

作者头像 李华