支持cuda的pytorch

(.venv) PS C:\Users\augus\PycharmProjects\pythonProject> pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
Looking in indexes: https://download.pytorch.org/whl/cu124
Requirement already satisfied: torch in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (2.4.0)
Collecting torchvision
Downloading https://download.pytorch.org/whl/cu124/torchvision-0.19.0%2Bcu124-cp312-cp312-win_amd64.whl (5.9 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.9/5.9 MB 3.1 MB/s eta 0:00:00
Collecting torchaudio
Downloading https://download.pytorch.org/whl/cu124/torchaudio-2.4.0%2Bcu124-cp312-cp312-win_amd64.whl (4.1 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.1/4.1 MB 4.1 MB/s eta 0:00:00
Requirement already satisfied: filelock in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (3.15.4)
Requirement already satisfied: typing-extensions>=4.8.0 in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (4.12.2)
Requirement already satisfied: sympy in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (1.13.2)
Requirement already satisfied: networkx in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (3.3)
Requirement already satisfied: jinja2 in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (3.1.4)
Requirement already satisfied: fsspec in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (2024.6.1)
Requirement already satisfied: setuptools in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torch) (72.2.0)
Requirement already satisfied: numpy in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torchvision) (1.26.4)
Collecting torch
Downloading https://download.pytorch.org/whl/cu124/torch-2.4.0%2Bcu124-cp312-cp312-win_amd64.whl (2503.3 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.5/2.5 GB 2.9 MB/s eta 0:00:00
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from torchvision) (10.4.0)
Requirement already satisfied: MarkupSafe>=2.0 in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from jinja2->torch) (2.1.5)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in c:\users\augus\pycharmprojects\pythonproject\.venv\lib\site-packages (from sympy->torch) (1.3.0)
Installing collected packages: torch, torchvision, torchaudio
Attempting uninstall: torch
Found existing installation: torch 2.4.0
Uninstalling torch-2.4.0:
Successfully uninstalled torch-2.4.0
Successfully installed torch-2.4.0+cu124 torchaudio-2.4.0+cu124 torchvision-0.19.0+cu124
(.venv) PS C:\Users\augus\PycharmProjects\pythonProject>
(.venv) PS C:\Users\augus\PycharmProjects\pythonProject>