Pytorch
- 9. Pytorchのインストール
• from os import path from wheel.pep425tags import get_abbr_impl,
get_impl_ver, get_abi_tag platform = ‘{}{}-{}’.format(get_abbr_impl(),
get_impl_ver(), get_abi_tag()) accelerator = ‘cu80’ if
path.exists(‘/opt/bin/nvidia-smi’) else ‘cpu’ !pip install -q
http://download.pytorch.org/whl/{accelerator}/torch-0.3.0.post4-
{platform}-linux_x86_64.whl torchvision
• # GPUの確認 import torch use_gpu = torch.cuda.is_available()
print(use_gpu)
• 上記のコマンドで、Trueが返ってくればインストール完了
- 10. Pytorchのインストール
• pip install -U torch torchvisionでもインストールできるかもしれません
が、各ライブラリのバージョンが違うとエラーが表示されるます。
• 詳しくはこちら
• Google ColaboratoryでPyTorch
• https://qiita.com/fukuit/items/342f5c8ec84e978fad3b
• Google Colaboratory事始め
• https://qiita.com/kouki_outstand/items/cd24dccbaa92274be39e