torchvisionとpytorchのインストール
310 ワード
conda install pytorch torchvision cuda92 -c pytorch
conda install pytorch cuda92 -c pytorch
特に注意:
バージョンは必ずパソコンのcudaバージョンに対応してください
完了したらpythonで表示できます
import torch
import torchvision
print(torch.__version__)# PyTorch