Skip to content

🔦 PyTorch

Some projects have Optional or Total Dependency on PyTorch

When inside the Virtual Environment, choose a version below and and run the Command

1. Select your Platform

Have the NVIDIA Drivers installed

Command
1
poe cuda

AMD doesn't support ROCm on Windows yet

It is supported on Linux, consider trying it there!

Help needed, I don't have the Hardware to test

Slow option, but works on any System

Command
1
poe cpu

Have the NVIDIA Proprietary Drivers packages installed in your Distro

Command
1
poe cuda

Have the Mesa Drivers and ROCm packages installed in your Distro

Command
1
poe rocm

Requires RX 5000 series or Newer. Might need to set HSA_OVERRIDE_GFX_VERSION

Should work, but I don't have the Hardware to test

Help needed, I don't have the Hardware to test

Slow option, but works on any System

Command
1
poe cpu

Command
1
poe base

Should work, but I don't have the Hardware to test