Pytorch 11.8

pixi.toml

[project]
name = "Pytorch"
version = "0.1.0"
description = "Add a short description here"
channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch"]
platforms = ["linux-64"]

[tasks]

[dependencies]
python = "3.11.*"
pip = ">=23.3.2,<23.4"
cuda = {version = "*", channel="nvidia/label/cuda-11.8.0"}
pytorch-cuda = {version = "11.8.*", channel="pytorch"}
pytorch = {version = ">=2.2.0,<2.3", channel="pytorch"}
torchvision = {version = ">=0.17.0,<0.18", channel="pytorch"}