ピトーチの公式ウェブサイトアドレス
現在、Pytorchバージョンの最新バージョンは1.4であり、
独自のCUDAバージョンに応じて最新のcuda9.2および10.1をサポートしています。適切なPytorchバージョンを選択できます
(自分のコンピューターまたはサーバーのCUDAバージョンに適応することを選択する必要があります。そうしないと、エラーが発生します!)
赤い文字をクリックして、次の画像にある以前のバージョンのPyTorchをインストールします。
これは
、次のURLです。https://pytorch.org/get-started/previous-versions/次のインターフェースを参照してください
。
ここでは、CUDA9.0とAnacondaに基づいてインストールしたため、選択したコマンドは次のとおりです。
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0
ここでは「-c pytorch」が削除されていることに注意してください。事前に構成されたドメスティックミラーソースが使用され、ダウンロード速度がはるかに速くなります。特定の構成方法は非常に単純ですが、繰り返します。
インストールが完了したら、次のように入力します。
python
anacondaのPython環境に
入り、次のように入力します。
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
次の画面が表示され
たら、もう一度入力できます。
print(torch.cuda.is_available())
ディスプレイ:真
インストール完了!