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

相关推荐
别叫我->学废了->lol在线等3 分钟前
python单例模式下线程安全优化
python·安全·单例模式
西江649766 分钟前
【个人博客系统—测试报告】
python·功能测试·jmeter·pycharm·postman
专注数据的痴汉7 分钟前
「数据获取」江门统计年鉴(1997-2024)
大数据·人工智能·信息可视化
小马爱打代码10 分钟前
Spring AI:文生视频 - wanx2.1-i2v-plus
java·人工智能·spring
小王毕业啦12 分钟前
2000-2023年 地级市-公路运输相关数据
大数据·人工智能·数据挖掘·数据分析·数据统计·社科数据·实证数据
象象翔19 分钟前
AI+若依(实战篇)
java·人工智能·spring boot·spring
Wishell201522 分钟前
日拱一卒之pytorch中的矩阵乘法
pytorch
月亮!24 分钟前
移动端测试重磅升级:跨平台自动化测试框架深度对比
运维·网络·人工智能·测试工具·容器·自动化·测试用例
CHANG_THE_WORLD25 分钟前
C++ vs Python 参数传递方式对比
java·c++·python
Mxsoft61928 分钟前
某次电缆绝缘预测不准,融合温湿度数据救场!
人工智能