Python distribution with PyTorch, Tensorflow, CUDA, and other libraries for running SPANets.
416
A python distribution based on Nvidia's CUDA container which contains the following python libraries
| Library | Included Version |
|---|---|
| python | 3.8 |
| numpy | 1.19.5 |
| sympy | 1.8 |
| scikit_learn | 0.24.2 |
| pytorch | 1.8.1 |
| pytorch-lightning | 1.2.10 |
| opt_einsum | 3.3.0 |
| h5py | 2.10 |
| numba | 0.53.1 |
We also include containers built on different versions of CUDA for driver compatility.
Content type
Image
Digest
Size
7.5 GB
Last updated
over 5 years ago
docker pull spanet/spanet-python