Ubuntu18.0へPyTorchのインストール

今日はPyTorchをインストール。

公式ページにはPyTorch（のみ）をpipやcondaでインストールする方法が

示されていたが、その前にCudaやCuDNNを先にインストールする必要がある。

私の環境は

OS：Ubuntu18.04

GPU：Geforce　GTX　1050

1.CUDA、CuDNNのインストール

$ sudo apt install nvidia-cuda-toolkit nvidia-driver-390

$ sudo reboot

$ echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" | sudo tee /etc/apt/sources.list.d/nvidia-ml.list

$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64/7fa2af80.pub

$ sudo apt update

$ sudo apt install libcudnn7-dev

2.PyTorchをインストール

pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

以上。PyTorchの方はsudoしてないけど、大丈夫だろうか。。

動作確認したらまた、アップします。