Pytorchのインストールと環境の構成

https://pytorch.org/get-started/previous-versions/

https://blog.csdn.net/thomaswu1992/article/details/90293015

condaを使用してインストールし、インストールする前にまずpytorchの仮想環境pytorch-py36を作成し、pythonはバージョン3.6を選択します

 

次に、Anacondaプロンプトでpytorch-py36をアクティブにしてこの環境に入り、次のコードをコピーして、Enterキーを押してインストールします。

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

 最後に、pytorch-py36環境で、pythonと入力し、次のコードを入力してインストールをテストします

import torch   # 导入torch包 如果没有报错则表示安装成功
torch.cuda.is_available()  # 返回True则表示GPU可用， 反之则不可用

CUDA9.0と対応するバージョンのcuDNNは、以前にtensorflowがインストールされたときにインストールされたため、このPytorchのインストールでは、NVIDIAの公式WebサイトからCUDAとcuDNNをダウンロードする必要はありませんでした。

ただし、インストール中に、数百メガバイトのcudatoolkit9.0をダウンロードしました。