Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Akihiro Nitta 2ab23f6af1 | 5 months ago | |
---|---|---|
.. | ||
Dockerfile | 10 months ago | |
README.md | 5 months ago | |
singularity | 2 years ago |
You can run PyG with CUDA 10.1 inside a docker image.
The creation of our dockerfile refers to the dockerfiles provided by NVIDIA and PyTorch.
$ docker build -t "custom image name"
$ docker run --rm -it --init --runtime=nvidia --ipc=host --network=host --volume=$PWD:/app -e NVIDIA_VISIBLE_DEVICES=0 "custom image name" /bin/bash
If you encounter any problems, please feel free to contact liangshengwen@ict.ac.cn.
You can run PyG inside a singularity image. An example singularity file can be found in this folder.
You might have to modify the script; depending on your needs, modify the following:
From: nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04
to the corresponding tag from https://hub.docker.com/r/nvidia/cuda. Same if you want to use anything but Ubuntu 16.04. Your host has to have at least this cuda version!pyenv install 3.7.2
and the following lines to the corresponding version.pip install torch==1.3.0
.pytorch_geometric
requirements (scatter: 1.4.0, sparse: 0.4.3, cluster: 1.4.5, geometric 1.3.2). To change these, change the corresponding git checkout
lines near the bottom./usr/local/cuda-10.1/extras/demo_suite/deviceQuery | grep 'CUDA Capability'
; if not, check here.Note: If your harddisk runs full after multiple builds, this is known and apparently working as intended; delete the /tmp/sbuild-XXXXXXXXX
files.
To build the container, run
sudo singularity build geometric.sif singularity
then wait. Once finished, you can run the GAT example in the folder you built the image in by calling
wget https://raw.githubusercontent.com/pyg-team/pytorch_geometric/master/examples/gat.py
(to download the sample),
then
singularity exec geometric.sif python3 gat.py
to run on the CPU, or
singularity exec --nv geometric.sif python3 gat.py
to run on the GPU.
No Description
Python CSV HTML+Django Shell other
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》