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

pytorch: remove pre-cuda92 hack

Browse Source
This commit is contained in:
Patrick Hilhorst 2018-10-31 10:43:00 +01:00
parent 4cfeb9da53
commit 43f6d9ada4
No known key found for this signature in database
GPG Key ID: 589BB0A8DAFEF2B2
1 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkgs/top-level/python-packages.nix
View File

@ -2282,20 +2282,8 @@ in {
grappelli_safe = callPackage ../development/python-modules/grappelli_safe { }; grappelli_safe = callPackage ../development/python-modules/grappelli_safe { };
pytorch = let pytorch = callPackage ../development/python-modules/pytorch {
# Fails with CUDA 9.1 and GCC 6.4:
# https://github.com/pytorch/pytorch/issues/5831
# https://devtalk.nvidia.com/default/topic/1028112
# We should be able to remove this when CUDA 9.2 is released.
cudatoolkit_9 = pkgs.cudatoolkit_9.override {
gcc6 = pkgs.gcc5;
};
in callPackage ../development/python-modules/pytorch {
cudaSupport = pkgs.config.cudaSupport or false; cudaSupport = pkgs.config.cudaSupport or false;
cudatoolkit = cudatoolkit_9;
cudnn = pkgs.cudnn_cudatoolkit_9.override {
inherit cudatoolkit_9;
};
}; };
pytorchWithCuda = self.pytorch.override { pytorchWithCuda = self.pytorch.override {