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