windows安装pytorch精简版(英伟达GPU)

1 下载anaconda

官网:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

选择下载Anaconda-1.4.0-Windows-x86.exe

2 创建虚拟环境

以管理员身份打开Anaconda Prompt

conda env list

conda creat -n yolov8 python=3.8

创建过程中有提示,填y即可

3 下载CUDA

前置条件:查看系统支持CUDA的版本

win+r,输入cmd回车,进入命令行输入nvidia-smi,再输入nvcc -V

cuda下载官网:CUDA 工具包档案 |NVIDIA 开发人员

4 安装pytorch

前置条件:配置镜像源

以管理员身份打开Anaconda Powershell Prompt

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

conda config --get

4.1官网获取下载命令

pytorch下载官网Start Locally | PyTorch

官网上复制命令

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

4.2下载pytorch

以管理员身份打开Anaconda Prompt

激活虚拟环境命令:

conda activate yolov8

下载pytorch命令:

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

输入conda list,看有没有pytorch或者torch

5 pycharm使用pytorch

相关推荐
夏天的味道٥几秒前
@JsonIgnore对Date类型不生效
开发语言·python
tsumikistep11 分钟前
【前后端】接口文档与导入
前端·后端·python·硬件架构
小毅&Nora21 分钟前
【人工智能】【AI外呼】系统架构设计与实现详解
人工智能·系统架构·ai外呼
小白学大数据44 分钟前
Python爬虫伪装策略:如何模拟浏览器正常访问JSP站点
java·开发语言·爬虫·python
jianqiang.xue1 小时前
别把 Scratch 当 “动画玩具”!图形化编程是算法思维的最佳启蒙
人工智能·算法·青少年编程·机器人·少儿编程
Coding茶水间2 小时前
基于深度学习的安全帽检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
头发还在的女程序员2 小时前
三天搞定招聘系统!附完整源码
开发语言·python
温轻舟2 小时前
Python自动办公工具06-设置Word文档中表格的格式
开发语言·python·word·自动化工具·温轻舟
weixin79893765432...2 小时前
Vue + Express + DeepSeek 实现一个简单的对话式 AI 应用
vue.js·人工智能·express
花酒锄作田2 小时前
[python]FastAPI-Tracking ID 的设计
python·fastapi