本文共 871 字，大约阅读时间需要 2 分钟。

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

直接从官网上下载对应版本的 。然后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

 

转载地址：https://blog.csdn.net/qq_23858101/article/details/84636678 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！