本文主要是介绍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
这篇关于windows安装pytorch精简版(英伟达GPU)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
