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