Comments (2)
I've created Dockerfile and it seems to work properly on K80 instance. Here it's:
FROM tobycheese/cuda:9.0-cudnn7-devel-ubuntu18.04
RUN apt-get update && apt-get install -y software-properties-common
RUN add-apt-repository ppa:deadsnakes/ppa -y
RUN apt-get update && apt-get install -y --no-install-recommends \
git \
tmux \
nano \
apt-utils \
python3.6 \
python3.6-dev \
python3-pip \
python3-setuptools \
g++-6 \
libsm6 \
libxrender1 \
libxtst6 \
&& \
rm -rf /var/lib/apt/lists/* && \
apt-get update
RUN update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-6 10 && \
update-alternatives --set g++ /usr/bin/g++-6
RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 10 && \
update-alternatives --set gcc /usr/bin/gcc-6
RUN git clone https://github.com/NVlabs/planercnn.git
RUN cd planercnn && pip3 install -r requirements.txt && pip3 install h5py torch==0.4.0 -f https://download.pytorch.org/whl/cu92/torch_stable.html
RUN gcc -v && cat /usr/local/cuda/version.txt
RUN ls /usr/local | grep cuda
RUN cd planercnn && \
cd nms/src/cuda/ && \
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -I /usr/local/cuda/include -x cu -Xcompiler -fPIC -arch=sm_37
RUN export LD_LIBRARY_PATH="/usr/local/cuda/include:$LD_LIBRARY_PATH" && \
cd planercnn/nms && \
sed -i 's|extra_objects=extra_objects|extra_objects=extra_objects,\n include_dirs=["/usr/local/cuda/include"]|' build.py && \
python3 build.py
RUN cd planercnn/roialign/roi_align/src/cuda && \
nvcc -c -o crop_and_resize_kernel.cu.o crop_and_resize_kernel.cu -I /usr/local/cuda/include -x cu -Xcompiler -fPIC -arch=sm_37
RUN cd planercnn/roialign/roi_align && \
export LD_LIBRARY_PATH="/usr/local/cuda-9.2/include:$LD_LIBRARY_PATH" && \
sed -i 's|extra_compile_args=extra_compile_args|extra_compile_args=extra_compile_args,\n include_dirs=["/usr/local/cuda/include"]|' build.py && \
python3 build.py
RUN pip3 install torch==0.4.1 -f https://download.pytorch.org/whl/cu92/torch_stable.html
However, when you're inside docker you'll have to rerun compilation and build of nms and roi_align as GPU is not visible to the while building docker.
So, just run:
cd planercnn
cd nms/src/cuda/
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=[arch]
cd ../../
python build.py
cd ../
cd roialign/roi_align/src/cuda/
nvcc -c -o crop_and_resize_kernel.cu.o crop_and_resize_kernel.cu -x cu -Xcompiler -fPIC -arch=[arch]
cd ../../
python build.py
cd ../../
from planercnn.
Hi @oOXpycTOo, I am using dockerfile created by you for "planercnn" project. I am having a system with 4 GPU's with Cuda 11.4.
While creating the image, I got following error for instruction RUN apt-get update && apt-get install -y software-properties-common:
W: GPG error: https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 InRelease: The following signatures couldn't be verified because the public key is not available: NO_PUBKEY A4B469963BF863CC
E: The repository 'https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 InRelease' is not signed.
The command '/bin/sh -c apt-get update && apt-get install -y software-properties-common' returned a non-zero code: 100
I have tried a lot of experimentation but could not resolve it. Kindly help me to get out of it.
Thanks.
from planercnn.
Related Issues (20)
- how can i get the normal of planes for each frame extracted from /scene_id/scene_id.sens? HOT 2
- Pretrained model links are not working HOT 3
- Nothing happened with nvcc HOT 6
- FileNotFoundError: [Errno 2] No such file or directory: '../../Data/ScanNet//scannetv2-labels.combined.tsv' HOT 5
- Pretrained model links are not working now HOT 2
- Torch summary on the model
- The ground truth provided by PlaneRCNN has large errors! HOT 2
- After training scannet from scratch while inferencing the trained model facing error HOT 1
- Plane parameter
- Unable to compile NMS and ROI-Align on RTX 3080 HOT 3
- Question about Training Data
- Keep the order of original image files while detecting planes
- plane_XYZ coordinate system
- cuda 8.0 on Ubuntu 18.0.4
- Plane segmentation evaluation experiment on ScanNet
- Trained Models
- Problem on download of scannet plane annotation
- How to find the plane normal? HOT 1
- How to transfer module from 'pth' to βonnxβοΌ eg: checkpoint_refine.pth to checkpoint_refine.onnx
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
π Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. πππ
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google β€οΈ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from planercnn.