torch不调用gpu

[Wang-Bo-xuan]

首先感谢您优秀的工作！

我在一台windows11上尝试训练，配置为32g + 2080s； cuda版本：

| NVIDIA-SMI 572.47 Driver Version: 572.47 CUDA Version: 12.8 |

nvcc版本：

Cuda compilation tools, release 12.8, V12.8.61

cudnn版本：

v9.7

使用anaconda维护虚拟环境；之前有尝试过几个数值回归的神经网络demo，安装的torch version=2.5.1；是可以调用gpu的； 在尝试本工程时，使用了requ文件，可以正常启动模型及对话，目前也可以进行预训练，但是无法调用gpu；想了解下如何解决？

虚拟环境内版本： python：

Python 3.10.16

pip：

pip 25.0 from D:\softwares\Anaconda\envs\minimind\lib\site-packages\pip (python 3.10)

pip list：

Package Version

---

accelerate 1.4.0 aiohappyeyeballs 2.4.6 aiohttp 3.11.12 aiosignal 1.3.2 altair 5.5.0 annotated-types 0.7.0 anyio 4.8.0 async-timeout 5.0.1 attrs 25.1.0 blinker 1.9.0 cachetools 5.5.2 certifi 2025.1.31 charset-normalizer 3.4.1 click 8.1.8 colorama 0.4.6 cycler 0.12.1 datasets 2.21.0 datasketch 1.6.4 dill 0.3.8 distro 1.9.0 docker-pycreds 0.4.0 exceptiongroup 1.2.2 filelock 3.17.0 Flask 3.0.3 Flask-Cors 4.0.0 fonttools 4.56.0 frozenlist 1.5.0 fsspec 2024.6.1 gitdb 4.0.12 GitPython 3.1.44 h11 0.14.0 httpcore 1.0.7 httpx 0.28.1 huggingface-hub 0.29.1 idna 3.10 importlib_metadata 7.2.1 itsdangerous 2.2.0 jieba 0.42.1 Jinja2 3.1.2 jiter 0.8.2 joblib 1.4.2 jsonlines 4.0.0 jsonschema 4.23.0 jsonschema-specifications 2024.10.1 kiwisolver 1.4.8 markdown-it-py 3.0.0 MarkupSafe 3.0.2 marshmallow 3.22.0 matplotlib 3.5.1 mdurl 0.1.2 mpmath 1.3.0 multidict 6.1.0 multiprocess 0.70.16 narwhals 1.27.1 networkx 3.4.2 ngrok 1.4.0 nltk 3.8 numpy 1.26.4 openai 1.59.6 packaging 23.2 pandas 1.5.3 peft 0.7.1 pillow 10.4.0 pip 25.0 platformdirs 4.3.6 propcache 0.3.0 protobuf 4.25.6 psutil 5.9.8 pyarrow 19.0.1 pydantic 2.8.2 pydantic_core 2.20.1 pydeck 0.9.1 Pygments 2.19.1 pyparsing 3.2.1 python-dateutil 2.9.0.post0 pytz 2025.1 PyYAML 6.0.2 referencing 0.36.2 regex 2024.11.6 requests 2.32.3 rich 13.7.1 rpds-py 0.23.1 safetensors 0.5.2 scikit-learn 1.5.1 scipy 1.15.2 sentence-transformers 2.3.1 sentencepiece 0.2.0 sentry-sdk 2.22.0 setproctitle 1.3.5 setuptools 75.8.0 simhash 2.1.2 six 1.17.0 smmap 5.0.2 sniffio 1.3.1 streamlit 1.30.0 sympy 1.13.3 tenacity 8.5.0 threadpoolctl 3.5.0 tiktoken 0.5.1 tokenizers 0.21.0 toml 0.10.2 torch 2.2.2 torchvision 0.17.2 tornado 6.4.2 tqdm 4.67.1 transformers 4.48.0 trl 0.13.0 typing_extensions 4.12.2 tzdata 2025.1 tzlocal 5.3 ujson 5.1.0 urllib3 2.3.0 validators 0.34.0 wandb 0.18.3 watchdog 6.0.0 Werkzeug 3.1.3 wheel 0.45.1 xxhash 3.5.0 yarl 1.18.3 zipp 3.21.0