目录结构与安装试用
1. 依赖
首先,需要一块 Nvidia 的显卡。
需要一个支持 C++14 的编译器,现在 Ubuntu 20.04 上通过 apt 安装的 GCC/G++ 都是 9.4 版本的 C++17 都是支持的。
sudo apt-get update
sudo apt-get install gcc g++
在 Linux 上,需要 10.2 以上的 CUDA,我按照教程安装了一个比较新的 12.2 版的CUDA。
需要 V3.21 以上版本的 CMake,这个安装起来稍微麻烦一点。因为 Ubuntu 20.04 通过 apt 默认安装的是 3.16 的,需要自己下载源码编译一下。 我们可以从这里找到历史发布的各个版本的CMake。编译CMake时,可能会提示 openssl 或者 curses 找不到,可以通过如下的指令解决:
sudo apt-get install libncurses5-dev
sudo apt-get install libssl-dev
然后就可以通过如下的指令下载-编译-安装cmake了。
# 以防万一,先把系统里面原有的 cmake 删掉
sudo apt-get remove cmake
wget https://cmake.org/files/v3.27/cmake-3.27.0-rc4.tar.gz
tar zxvf cmake-3.27.0-rc4.tar.gz
cd cmake-3.27.0-rc4
./bootstrap
make
sudo make install
如果一切顺利,可以通过如下指令查看 cmake 的版本。
cmake --version
2. 编译 & 试用
git clone --recursive https://github.com/gaoyichao/tiny-cuda-nn.git
cd tiny-cuda-nn
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=RelWithDebInfo ..
make -j8