为了能让colab操做google driver中的文件须要执行下面代码。python
!apt-get install -y -qq software-properties-common python-software-properties module-init-tools !add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null !apt-get update -qq 2>&1 > /dev/null !apt-get -y install -qq google-drive-ocamlfuse fuse from google.colab import auth auth.authenticate_user() from oauth2client.client import GoogleCredentials creds = GoogleCredentials.get_application_default() import getpass !google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL vcode = getpass.getpass() !echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}
中途会让你输入两次受权码。下面是执行成功界面的截图。
linux
而后在google driver中用建立一个colab的ipynb文件。
而后你须要指定你之后在程序下载的东西存储在google driver 哪里,选择Files
而后选择MOUNT DRIVE
。
而后能够看到你的代码变到了某个磁盘下的某个文件夹下了,之后你保存的东西也在这里。
而后切换当前工做目录到你想要的那个目录下。我是切换到train_pvnet那个目录而后下载代码。
而后从github下载PVNet的源码。输入下面这个命令而后按ctr+enter
运行命令,注意感叹号是指运行linux命令。
!git clone https://github.com/zju3dv/pvnet.git
而后将工做目录切换刚刚下载好的那个项目的文件夹pvnet
下。
%cd pvnet
你能够经过输入命令!pwd
查看当前目录,也能够输入命令!ls
看看当前目录下有哪些文件。
而后再在google driver中的那个pvnet文件夹下建立一个colab文件方便咱们执行训练PVNet的训练。
具体训练过程须要参考他们在github给的readme文件。git
CUDA_HOME='/usr/local/cuda' %cd /content/drive/My\ Drive/train_pvnet/pvnet/lib/ransac_voting_gpu_layer !python setup.py build_ext --inplace
记得将runtime里面的加速设备设置为GPU。
而后再执行命令github
# 修订cuda_include并dart在build_extend_utils_cffi.py为与CUDA在您的电脑兼容 %cd /content/drive/My\ Drive/train_pvnet/pvnet/lib/utils/extend_utils !sudo apt-get install libgoogle-glog-dev !sudo apt-get install libsuitesparse-dev !sudo apt-get install libatlas-base-dev !python3 build_extend_utils_cffi.py
接下来咱们须要安装ceres这个数学工具库,这里咱们提供的是下载源码进行编译。在安装编译它以前须要安装着libeigen3-dev,libgoogle-glog-dev这两个包。
而后再运行pvnet这个目录下的build_ceres.sh。而后移动ceres/ceres-solver/build/lib/libceres.so*到lib/utils/extend_utils/lib。web