15 lines
587 B
Nix
15 lines
587 B
Nix
|
{
|
||
|
|
x86_64-darwin-cpu = {
|
||
|
|
url = "https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.6.0.zip";
|
||
|
|
sha256 = "0d4n7la31qzl4s9pwvm07la7q6lhcwiww0yjpfz3kw6nvx84p22r";
|
||
|
|
};
|
||
|
|
x86_64-linux-cpu = {
|
||
|
|
url = "https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.6.0%2Bcpu.zip";
|
||
|
|
sha256 = "1975b4zvyihzh89vnwspw0vf9qr05sxj8939vcrlmv3gzvdspcxz";
|
||
|
|
};
|
||
|
|
x86_64-linux-cuda = {
|
||
|
|
url = "https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.6.0.zip";
|
||
|
|
sha256 = "127qnfyi1faqbm40sbnsyqxjhrqj82bzwqyz7c1hs2bm0zgrrpya";
|
||
|
|
};
|
||
|
|
}
|