您可以尝试安装 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
您可以尝试安装 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