This repo provides a guide to set up the GPU environment for deep learning frameworks (TensorFlow, PyTorch) on the Google Cloud Platform.
-
Updated
Sep 26, 2020
#
gpu-setup
Here are 4 public repositories matching this topic...
pytorch tensorflow-tutorials cudnn cuda-toolkit tensorflow-gpu deeplearning-framework conda-env tensorflow-setup pytorch-implementation gpu-setup
-
Updated
Aug 28, 2025 - Jupyter Notebook
Universal GPU setup for PyTorch and JAX — DirectML, ROCm, CUDA, MPS, CPU. Auto-detects hardware, fixes HSA_OVERRIDE_GFX_VERSION for AMD RX 5700 XT. Beats devicetorch + torchruntime combined.
windows macos linux deep-learning cross-platform amd gpu cuda pytorch mps rocm jax gpu-detection wsl2 gpu-setup directml rx-5700-xt gfx1010 hsa-override get-best-device
-
Updated
Mar 28, 2026 - Python
GPU-Enabled Development Environment Demo for Gitpod Flex
-
Updated
Sep 10, 2025 - Jupyter Notebook
Improve this page
Add a description, image, and links to the gpu-setup topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the gpu-setup topic, visit your repo's landing page and select "manage topics."