Ubuntu 16.04でpytorchをインストールするプロセス全体

ピトーチの公式ウェブサイトアドレス

現在、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())

ディスプレイ:真

インストール完了!

オリジナルの記事を2件公開しました Likes0 Visits12

おすすめ

転載: blog.csdn.net/Jaredzzz/article/details/105589094