Skip to content

Conversation

@mattshax
Copy link
Member

Summary

  • Documents that ray[default] doesn't include deep learning libraries
  • Adds instructions for installing PyTorch/TensorFlow in the Ray venv
  • Clarifies that gpu_workload.py shows "PyTorch not available" unless installed

Test plan

  • Verify README renders correctly

Clarifies that ray[default] doesn't include deep learning libraries and
shows how to install PyTorch/TensorFlow for GPU workloads.
Copilot AI review requested due to automatic review settings January 30, 2026 22:55
Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds documentation to help users install deep learning libraries (PyTorch and TensorFlow) in their Ray virtual environment, since the default ray[default] installation doesn't include them.

Changes:

  • Added a new "Installing Additional Packages" section explaining how to install PyTorch or TensorFlow
  • Clarified that examples/gpu_workload.py works without PyTorch but displays "PyTorch not available" messages unless PyTorch is installed

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment on lines +75 to +76
# Or install TensorFlow
uv pip install --python ~/pw/activate-ray/ray_venv/bin/python tensorflow
Copy link

Copilot AI Jan 30, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The comment says "Install PyTorch with CUDA support" but the command pip install torch will install the CPU-only version by default. To install PyTorch with CUDA support, you need to specify the appropriate PyTorch index URL or follow PyTorch's installation instructions for CUDA-enabled versions. Consider either:

  1. Updating the command to include CUDA support (e.g., using PyTorch's recommended installation command with the correct index URL)
  2. Changing the comment to clarify that this installs the default version and users should consult PyTorch documentation for CUDA-specific installations
  3. Removing the "with CUDA support" phrase from the comment if the intention is just to install the default version

Copilot uses AI. Check for mistakes.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants