天天看點

Torch 完美安裝安裝Torch安裝cuda設定cuda環境變量安裝torch的cuda支援下載下傳安裝cudnn(注意與cuda版本對應)安裝torch的cudnn支援

Ubuntu Terminal 安裝指令:

安裝Torch

# in a terminal, run the commands WITHOUT

sudo git clone https://github.com/torch/distro.git ~/torch --recursive

cd ~/torch; bash install-deps;

./install.sh

安裝cuda

直接從官網上下載下傳對應版本的cuda 。然後deb檔案直接輕按兩下安裝。你别以為這個deb安裝完了,cuda就安裝好了,其實還有附加的庫,也是要安裝的。這些要通過指令行來進行。

sudo apt-get update

sudo apt-get install cuda

設定cuda環境變量

打開終端輸入以下指令:

sudo gedit /etc/bash.bashrc

2.在配置檔案中輸入以下cuda環境變量:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export CUDA_HOME=/usr/local/cuda

儲存退出。

3.source /etc/bash.bashrc 立即生效

安裝torch的cuda支援

luarocks install cutorch

luarocks install cunn

下載下傳安裝cudnn(注意與cuda版本對應)

tar xvzf cudnn-8.0-linux-x64-v6.0-tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

安裝torch的cudnn支援

luarocks install cudnn