gitee导航
直接下载自己改,源码我指不定哪天就会按我自己的训练修改
1.gpu环境安装
cmd执行
nvidia-smi

主要看下cuda的版本的显存大小

上面标记的是我下载下来能跑起来的版本
python 选择 Windows installer (64-bit),下载安装,勾选添加环境
地址:https://www.python.org/downloads/release/python-3116/
cmd查看是否有pip或者pip3, 没有的话cmd执行
python -m ensurepip --upgrade
cuda安装,选个自己显卡支持的版本,全部点下一步,装C盘蟹蟹!!
https://developer.nvidia.com/cuda-toolkit-archive
torch官网安装,一般会很慢,要么翻墙要么自己先去下面的http链接下载,全局安装就行了
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
本文介绍了如何在Windows上安装GPU环境,包括检查CUDA版本、安装Python、确保pip存在并升级,以及使用PyTorch的官方和镜像站快速全局安装torch、torchvision和torchaudio的过程。

1万+

被折叠的 条评论
为什么被折叠?



