1.準備
（1）nvidiaドライバーのインストール：ubuntu16深層学習環境の構築（1）
-nvidiaドライバーのインストール（2）cudaとcudnnのインストール：ubuntu16深層学習環境の構築（2）-cudaとcudnnの
インストール（3）Anacondaのインストール：ubuntu16のインストールアナコンダ3
2.仮想環境を作成します
最初にミラーソースを追加します。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
仮想環境を再度作成します。
conda create -n pytorchgpu python=3.7 //pytorchgpu是自己设置的环境名字
conda activate pytorchgpu //激活环境
3.pytorchをインストールします
次の図に示すように、最初にpytorchの公式Webサイトhttps://pytorch.org/にアクセスして、対応するバージョンを検索します。
指定されたコマンドを貼り付けるだけです。
4.cudaとcudnnがpytorchで利用できるかどうかを確認します
ターミナルにpythonと入力し、次のコードを入力します。
import torch
print（torch.cuda.is_available（））// torch.backends
からcudaをインストールした場合はTrueを返しますimportcudnn
print（cudnn.is_available（））// cudaをインストールした場合はTrueを返します