search

h2oai / driverlessai-recipes

Recipes for Driverless AI
Apache License 2.0
240 stars 93 forks source link

Don't use AMD ROCm variant of `torchvision` package #200

Closed rsujeevan closed 5 months ago

rsujeevan commented 5 months ago

ROCm is for AMD GPUs and it's not available from pypi