按照课程提示:安装了 cuda 版本为:11.8, cudnn 也安装成功
安装 pytorch:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
运行代码报警告
D:\Program\envs\L2_01\lib\site-packages\torch\cuda\__init__.py:287: UserWarning:
NVIDIA GeForce RTX 5060 Laptop GPU with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90 compute_37.
If you want to use the NVIDIA GeForce RTX 5060 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/