search

hasktorch / hasktorch-skeleton

Nix Skeleton Hasktorch project made for easy cloning and forking
MIT License
14 stars 18 forks source link

Fix CUDA 11 version to match libtorch-nix #8

Open Reep236 opened 3 years ago

Reep236 commented 3 years ago

libtorch_cudatoolkit_11_0 -> libtorch_cudatoolkit_11_1

Attempting to use nix with cudaMajorVersion 11 would fail with a missing attribute error as libtorch-nix has 11.1 instead of 11.0