PS C:\Users\user> pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 –extra-index-url https://download.pytorch.org/whl/cu117
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu117
Collecting torch==1.13.1+cu117
Downloading https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp310-cp310-win_amd64.whl (2255.4 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.3/2.3 GB 3.6 MB/s eta 0:00:00
Collecting torchvision==0.14.1+cu117
Downloading https://download.pytorch.org/whl/cu117/torchvision-0.14.1%2Bcu117-cp310-cp310-win_amd64.whl (4.8 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.8/4.8 MB 155.8 MB/s eta 0:00:00
Collecting torchaudio==0.13.1
Downloading https://download.pytorch.org/whl/cu117/torchaudio-0.13.1%2Bcu117-cp310-cp310-win_amd64.whl (2.3 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.3/2.3 MB 73.3 MB/s eta 0:00:00
Collecting typing-extensions (from torch==1.13.1+cu117)
Downloading typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Collecting numpy (from torchvision==0.14.1+cu117)
Downloading numpy-1.24.3-cp310-cp310-win_amd64.whl (14.8 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 14.8/14.8 MB 81.8 MB/s eta 0:00:00
Collecting requests (from torchvision==0.14.1+cu117)
Downloading requests-2.29.0-py3-none-any.whl (62 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 62.5/62.5 kB ? eta 0:00:00
Collecting pillow!=8.3.*,>=5.3.0 (from torchvision==0.14.1+cu117)
Downloading Pillow-9.5.0-cp310-cp310-win_amd64.whl (2.5 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.5/2.5 MB 166.7 MB/s eta 0:00:00
Collecting charset-normalizer<4,>=2 (from requests->torchvision==0.14.1+cu117)
Downloading charset_normalizer-3.1.0-cp310-cp310-win_amd64.whl (97 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 97.1/97.1 kB ? eta 0:00:00
Collecting idna<4,>=2.5 (from requests->torchvision==0.14.1+cu117)
Downloading https://download.pytorch.org/whl/idna-3.4-py3-none-any.whl (61 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.5/61.5 kB ? eta 0:00:00
Collecting urllib3<1.27,>=1.21.1 (from requests->torchvision==0.14.1+cu117)
Downloading urllib3-1.26.15-py2.py3-none-any.whl (140 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 140.9/140.9 kB ? eta 0:00:00
Collecting certifi>=2017.4.17 (from requests->torchvision==0.14.1+cu117)
Downloading https://download.pytorch.org/whl/certifi-2022.12.7-py3-none-any.whl (155 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 155.3/155.3 kB ? eta 0:00:00
Installing collected packages: urllib3, typing-extensions, pillow, numpy, idna, charset-normalizer, certifi, torch, requests, torchvision, torchaudio
Successfully installed certifi-2022.12.7 charset-normalizer-3.1.0 idna-3.4 numpy-1.24.3 pillow-9.5.0 requests-2.29.0 torch-1.13.1+cu117 torchaudio-0.13.1+cu117 torchvision-0.14.1+cu117 typing-extensions-4.5.0 urllib3-1.26.15
PS C:\Users\user>
PS C:\Users\user> pip list
Package Version
certifi 2022.12.7
charset-normalizer 3.1.0
idna 3.4
numpy 1.24.3
Pillow 9.5.0
pip 23.1.2
requests 2.29.0
setuptools 65.5.0
torch 1.13.1+cu117
torchaudio 0.13.1+cu117
torchvision 0.14.1+cu117
typing_extensions 4.5.0
urllib3 1.26.15
PS C:\Users\user>
PS C:\Users\user> pip show torch
Name: torch
Version: 1.13.1+cu117
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: c:\users\user\appdata\local\programs\python\python310\lib\site-packages
Requires: typing-extensions
Required-by: torchaudio, torchvision
PS C:\Users\user>
PS C:\Users\user> py C:\Users\user/python_exercise/torch_gpu_info.py
torch.version, 1.13.1+cu117
torch.cuda.is_available(), True
compute_89
find gpu devices, 1
cuda:0, NVIDIA GeForce RTX 4090
end
PS C:\Users\user>
PS C:\Users\user\python_exercise> cat torch_gpu_info.py
import torch
print(f”torch.version, {torch.version}”)
print(f”torch.cuda.is_available(), {torch.cuda.is_available()}”)
print(f”compute_{”.join(map(str,(torch.cuda.get_device_capability())))}”)
device_num:int = torch.cuda.device_count()
print(f”find gpu devices, {device_num}”)
for idx in range(device_num):
print(f”cuda:{idx}, {torch.cuda.get_device_name(idx)}”)
print(“end”)
PS C:\Users\user\python_exercise>