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

相关推荐
小兜全糖(xdqt)4 分钟前
python ppt转pdf以及图片提取
python·pdf·powerpoint
互联科技报5 分钟前
AI赋能企业办公:文多多AiPPT以技术创新破解行业痛点
人工智能
番石榴AI9 分钟前
视频转ppt/pdf V2.0版(新增转为可编辑PPT功能)
人工智能·pdf·powerpoint
前端世界14 分钟前
用Python打造智能成绩分析系统:从异常处理到断言验证的全流程实战
服务器·数据库·python
yaoxin52112314 分钟前
229. Java 集合 - 操作集合中的多个元素(批量操作)
java·开发语言·python
岁岁岁平安15 分钟前
python 配置pip镜像源
python
在人间负债^17 分钟前
从Python到仓颉:核心项目内容迁移实践
开发语言·python·鸿蒙·仓颉
c0d1ng19 分钟前
自建督学习——BERT(第二十二周周报)
人工智能·学习·bert
云和数据.ChenGuang19 分钟前
SyntaxError: Non-UTF-8 code starting
人工智能·python·numpy
站大爷IP37 分钟前
Airflow调度爬虫任务:从零搭建高效定时采集系统
python