Pytorch pipインストール

機械学習のために、pytorchを入れたく、https://pytorch.org/get-started/locally/
を参照に実行してみました。
PyTorch Build:Stable (1.13.1)
Your OS:Windows
Package:Pip
Language:Python
Compute Platform:CUDA 11.7

コードは

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

torch-1.12.1と異なるバージョンが入るとともに、CUDAは入っていないように見えます

(base) C:\Users\admin\Desktop\yolov5>pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu117 Collecting torch Using cached https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp39-cp39-win_amd64.whl (2255.6 MB) Requirement already satisfied: torchvision in c:\users\admin\anaconda3\lib\site-packages (0.13.1) Requirement already satisfied: torchaudio in c:\users\admin\anaconda3\lib\site-packages (0.12.1) Requirement already satisfied: typing-extensions in c:\users\admin\anaconda3\lib\site-packages (from torch) (4.1.1) Using cached torch-1.12.1-cp39-cp39-win_amd64.whl (161.8 MB) Requirement already satisfied: numpy in c:\users\admin\anaconda3\lib\site-packages (from torchvision) (1.23.1) Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in c:\users\admin\anaconda3\lib\site-packages (from torchvision) (9.2.0) Requirement already satisfied: requests in c:\users\admin\anaconda3\lib\site-packages (from torchvision) (2.28.1) Requirement already satisfied: urllib3<1.27,>=1.21.1 in c:\users\admin\anaconda3\lib\site-packages (from requests->torchvision) (1.26.9) Requirement already satisfied: idna<4,>=2.5 in c:\users\admin\anaconda3\lib\site-packages (from requests->torchvision) (3.3) Requirement already satisfied: certifi>=2017.4.17 in c:\users\admin\anaconda3\lib\site-packages (from requests->torchvision) (2021.10.8) Requirement already satisfied: charset-normalizer<3,>=2 in c:\users\admin\anaconda3\lib\site-packages (from requests->torchvision) (2.0.4) Installing collected packages: torch Successfully installed torch-1.12.1

どうすれば、torch==1.13.1+cu117を入れれるでしょうか。

コメントを投稿

0 コメント