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

相关推荐
IT_陈寒6 分钟前
React的useState居然还有这种坑?我差点删库跑路
前端·人工智能·后端
用户413062258291 小时前
给AI回答加引用角标citation:RAG前端实现
人工智能
金銀銅鐵1 小时前
[Python] 从《千字文》中随机挑选汉字
后端·python
米小虾1 小时前
WAIC 2026 倒计时30天:300+ AI 产品全球首发,今年看点全解析
人工智能
码上天下2 小时前
多模态Agent上传图片:前端压缩格式与预览实战
人工智能
姗姗来迟了2 小时前
Vue3封装可复用AI对话组件:一次抽象复盘
人工智能
怕浪猫2 小时前
哪些软件对 Chrome DevTools Protocol 频繁使用
人工智能·架构·前端框架
leo在掘金4 小时前
从DeepSeek 510亿融资到GitHub 33K Star开源项目:这周的技术生态发生了什么?
人工智能