bushlab/pytorch repository overview

The latest version is v1.6.

This container comes with scikit-learn, Tensorflow, Jupyter Lab and Pytorch installed, with CUDA 11.3 support, along with NVIDIA driver 470.57.02 for Linux x86_64, and has GPU support for NVIDIA GPUs with CUDA 11.3.

⁠Running the container

Run docker run -it -p 8888:8888 bushlab/pytorch:v1.6.

⁠Launching Jupyter lab

Once inside the container, run jupyter lab --allow-root --ip=0.0.0.0 --port=8888.

⁠Opening Jupyter lab

Enter local-host:8888 into a browser window and copy the token to use Jupyter lab.

⁠Disclaimer

CUDA has been installed, but the version is 11.3. Please be aware!

v1.3 and above come with NVIDIA driver version 470.57.02 for Linux x86_64 installed.

This container has been confirmed to work with GPUs compatible with NVIDIA driver version 470.57.02, alongside CUDA version 11.3.