环境配置

PyTorch

https://pytorch.org/get-started/locally/

去这里

看cuda版本

nvidia-smi

安装

然后复制对应的pip命令

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu129

SKLearn

安装

pip install scikit-learn