Ubuntu18.0へPyTorchのインストール
今日はPyTorchをインストール。
公式ページにはPyTorch(のみ)をpipやcondaでインストールする方法が
示されていたが、その前にCudaやCuDNNを先にインストールする必要がある。
私の環境は
OS:Ubuntu18.04
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してないけど、大丈夫だろうか。。
動作確認したらまた、アップします。