在这里插入图片描述
您可以尝试安装 CUDA 12.1 驱动程序,或者使用与 CUDA 12.1 兼容的 PyTorch 2.4.1。请使用以下命令来安装 PyTorch 2.4.1+cu121:

pip install torch==2.4.1+cu121 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121


在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐