Skip to content

Commit 37094b6

Browse files
author
Yuke Wang
committed
update README
1 parent 212414e commit 37094b6

1 file changed

Lines changed: 18 additions & 1 deletion

File tree

README.md

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,27 @@ git clone --recursive git@github.com:YukeWang96/OSDI21_AE.git
3838
### **Step-1: Environment Setup**
3939
There are two ways to setup the environment of GNNAdvisor and baselines.
4040
### + **Method 1**: Setup the environment via Docker (**Recommended**).
41-
+ Install Docker Engine with NVIDIA GPU Support **[Toturial](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker)**.
41+
+ Install Docker Engine with NVIDIA GPU Support **[Toturial](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker)**. We use the following commands
42+
```
43+
curl https://get.docker.com | sh \
44+
&& sudo systemctl --now enable docker
45+
46+
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
47+
&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
48+
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
49+
50+
sudo apt-get update
51+
sudo apt-get install -y nvidia-docker2
52+
sudo systemctl restart docker
53+
```
4254
+ `cd Docker` then either goto `cu102/` (for Quadro P6000 and Tesla V100) or `cu110/` (for RTX3090/RTX3070).
4355
+ Run `./build.sh`, it may takes a while (around 10 minutes) for building the container.
4456
+ Run `./launch.sh` then it will bring up an new interactive command line interface.
57+
> + if your enounter problem below, most likely, you may not install the GPU docker engine correctly, you may go back to the Toturial at the first setp.
58+
59+
```
60+
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
61+
```
4562
> + The defualt GPU is device:0. If you want to run on different deivce. Consider using this command in `./launch.sh`, e.g., using device:1
4663
>
4764
```

0 commit comments

Comments
 (0)