web-dev-qa-db-ja.com

Google ColabにPyTorch v1.0.0 +をインストールするにはどうすればよいですか？

PyTorch v1.0.0安定版は 2018年12月8日にリリース であった後 7か月前に発表 .

IPythonカーネルが実行されているハードウェア用に最適化されたバージョンを取得したい。

Google Colabでこのバージョンを入手するにはどうすればよいですか？

pytorchgoogle-colaboratory
7
Tom Hale

次のコードスニペットを試してください（gpuの有無に関係なく、ランタイムでも同様に機能します）

!pip install -q torch==1.0.0 torchvision

バージョンを確認するには

import torch
print(torch.__version__)

ここにバージョン1.0.0があります

[〜＃〜] update [〜＃〜]

!pip install torch

最も安定したバージョンは1.0.0

9
ashunigion

バージョン1.0.0では、PyTorchはダウンロードURLの形式を次から変更しました：

https://download.pytorch.org/whl/cu92/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

に

https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

（変更はCUDAバージョン部分にあり、cu92はcu90に変更されます。）

プログラムでそのURLを生成するために、次のコードを使用しました。

from os.path import exists
from wheel.pep425tags import get_abbr_impl, get_impl_ver, get_abi_tag

platform = '{}{}-{}'.format(get_abbr_impl(), get_impl_ver(), get_abi_tag())
cuda_output = !ldconfig -p|grep cudart.so|sed -e 's/.*\.\([0-9]*\)\.\([0-9]*\)$/cu\10/'    
accelerator = cuda_output[0] if exists('/dev/nvidia0') else 'cpu'

torch_url=f"http://download.pytorch.org/whl/{accelerator}/torch-{version}-{platform}-linux_x86_64.whl"
version='1.0.0'

!pip install -U {torch_url} torchvision

新しいバージョンがリリースされると、必要に応じてversion変数を変更できます。

3
Tom Hale

バージョン1.1.0の場合、これは機能します

!pip install -q torch==1.1.0 torchvision
2
Dimeji Mudele

今すぐできる

import torch

追加インストールの必要はありません。

1
Lê Vũ Linh