gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.torchvision: 0.25.0 -> 0.26.0

Browse Source 
Diff: https://github.com/pytorch/vision/compare/v0.25.0...v0.26.0

Changelog: https://github.com/pytorch/vision/releases/tag/v0.26.0
This commit is contained in:
Gaetan Lepage
2026-03-27 21:57:36 +00:00
parent 177911f865
commit d001cf04cb
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torchvision/default.nix
+15 -3
View File
@@ -26,16 +26,16 @@ let
inherit (torch) cudaCapabilities cudaPackages cudaSupport;
in
buildPythonPackage.override { stdenv = torch.stdenv; } (finalAttrs: {
buildPythonPackage.override { inherit (torch) stdenv; } (finalAttrs: {
pname = "torchvision";
version = "0.25.0";
version = "0.26.0";
pyproject = true;
src = fetchFromGitHub {
owner = "pytorch";
repo = "vision";
tag = "v${finalAttrs.version}";
hash = "sha256-oktJHcT6T4f58pUO+HSBpbyS1ISH3zDlTsXQh6PcMy4=";
hash = "sha256-FOdDGY3v8yWBhtNo9tZP79/xwrc7AoIY5Y1ZABzWe6g=";
};
nativeBuildInputs = [
@@ -63,6 +63,18 @@ buildPythonPackage.override { stdenv = torch.stdenv; } (finalAttrs: {
TORCHVISION_LIBRARY = "${libjpeg_turbo}/lib/";
}
// lib.optionalAttrs cudaSupport {
# At of 2026-03-27, the default `cudaPackages` version is 12.9.1
# According to Nvidia, it should support GCC versions up to 14.x:
# -> https://docs.nvidia.com/cuda/archive/12.9.1/cuda-installation-guide-linux/index.html#host-compiler-support-policy
# However, PyTorch's *strict* upper bound is 14.0:
# -> https://github.com/pytorch/pytorch/blob/v2.11.0/torch/utils/cpp_extension.py#L75
# Hence, the build fails with:
# RuntimeError: The current installed version of g++ (14.3.0) is greater than the maximum
# required version by CUDA 12.9. Please make sure to use an adequate version of g++
# (>=6.0.0, <14.0).
# Hence, we disable the version check to silence the error:
TORCH_DONT_CHECK_COMPILER_ABI = 1;
TORCH_CUDA_ARCH_LIST = "${lib.concatStringsSep ";" cudaCapabilities}";
FORCE_CUDA = 1;
};