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

相关推荐
小关会打代码几秒前
计算机视觉案例分享之答题卡识别
人工智能·计算机视觉
Se7en25817 分钟前
使用 NVIDIA Dynamo 部署 PD 分离推理服务
人工智能
海拥24 分钟前
用 LazyLLM 搭建一个代码注释 / 文档 Agent 的实测体验
人工智能
天天进步201525 分钟前
用Python打造专业级老照片修复工具:让时光倒流的数字魔法
人工智能·计算机视觉
liwulin050632 分钟前
【ESP32-CAM】HELLO WORLD
python
文火冰糖的硅基工坊33 分钟前
《投资-54》数字资产的形式有哪些?
人工智能·区块链
机器之心36 分钟前
刚刚,OpenAI发布GPT-5-Codex:可独立工作超7小时,还能审查、重构大型项目
人工智能·openai
荼蘼38 分钟前
答题卡识别改分项目
人工智能·opencv·计算机视觉
Doris_20231 小时前
Python条件判断语句 if、elif 、else
前端·后端·python
Doris_20231 小时前
Python 模式匹配match case
前端·后端·python