search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.46k stars 3.68k forks source link

updated Dockerfile based on NGC PyG 24.09 image #9794

Open sbhavani opened 1 week ago

sbhavani commented 1 week ago

Updated to use new NGC CUDA DL base image. Some differences:

  1. /workspace is the working directory
  2. Python libs removed that were not included in NGC PyG image: torch_scatter torch_sparse torch_cluster torch_spline_conv torchnet==0.0.4 h5py torchnet
  3. Using latest stable versions for graphviz and torch
puririshi98 commented 4 days ago

will merge once @sbhavani has tests passing locally