···250250 symbol_level = 0;
251251 fieldtrial_testing_like_official_build = true;
252252253253- # Google API keys, see:
254254- # http://www.chromium.org/developers/how-tos/api-keys
255255- # Note: These are for NixOS/nixpkgs use ONLY. For your own distribution,
256256- # please get your own set of keys.
253253+ # Google API key, see: https://www.chromium.org/developers/how-tos/api-keys
254254+ # Note: The API key is for NixOS/nixpkgs use ONLY.
255255+ # For your own distribution, please get your own set of keys.
257256 google_api_key = "AIzaSyDGi15Zwl11UNe6Y-5XW_upsfyw31qwZPI";
258258- google_default_client_id = "404761575300.apps.googleusercontent.com";
259259- google_default_client_secret = "9rIFQjfnkykEmqb6FfjJQD1D";
260257 } // optionalAttrs proprietaryCodecs {
261258 # enable support for the H.264 codec
262259 proprietary_codecs = true;
···1818 # this derivation. However, we should ensure on version bumps
1919 # that the CUDA toolkit for `passthru.tests` is still
2020 # up-to-date.
2121- version = "1.7.1";
2121+ version = "1.8.0";
2222 device = if cudaSupport then "cuda" else "cpu";
2323 srcs = import ./binary-hashes.nix version;
2424 unavailable = throw "libtorch is not available for this platform";
···7474 # (allowing FBGEMM to be built in pytorch-1.1), and may future proof this
7575 # derivation.
7676 brokenArchs = [ "3.0" ]; # this variable is only used as documentation.
7777- cuda9ArchList = [
7878- "3.5"
7979- "5.0"
8080- "5.2"
8181- "6.0"
8282- "6.1"
8383- "7.0"
8484- "7.0+PTX" # I am getting a "undefined architecture compute_75" on cuda 9
8585- # which leads me to believe this is the final cuda-9-compatible architecture.
8686- ];
8787- cuda10ArchList = cuda9ArchList ++ [
8888- "7.5"
8989- "7.5+PTX" # < most recent architecture as of cudatoolkit_10_0 and pytorch-1.2.0
9090- ];
7777+7878+ cudaCapabilities = rec {
7979+ cuda9 = [
8080+ "3.5"
8181+ "5.0"
8282+ "5.2"
8383+ "6.0"
8484+ "6.1"
8585+ "7.0"
8686+ "7.0+PTX" # I am getting a "undefined architecture compute_75" on cuda 9
8787+ # which leads me to believe this is the final cuda-9-compatible architecture.
8888+ ];
8989+9090+ cuda10 = cuda9 ++ [
9191+ "7.5"
9292+ "7.5+PTX" # < most recent architecture as of cudatoolkit_10_0 and pytorch-1.2.0
9393+ ];
9494+9595+ cuda11 = cuda10 ++ [
9696+ "8.0"
9797+ "8.0+PTX" # < CUDA toolkit 11.0
9898+ "8.6"
9999+ "8.6+PTX" # < CUDA toolkit 11.1
100100+ ];
101101+ };
91102 final_cudaArchList =
92103 if !cudaSupport || cudaArchList != null
93104 then cudaArchList
9494- else
9595- if lib.versions.major cudatoolkit.version == "9"
9696- then cuda9ArchList
9797- else cuda10ArchList; # the assert above removes any ambiguity here.
105105+ else cudaCapabilities."cuda${lib.versions.major cudatoolkit.version}";
9810699107 # Normally libcuda.so.1 is provided at runtime by nvidia-x11 via
100108 # LD_LIBRARY_PATH=/run/opengl-driver/lib. We only use the stub