diff --git a/conda/meta.yaml b/conda/meta.yaml index b57a64bf820c7063b05ec0e351eb3576c77c9000..034b94053562454ffa99354c9b8a42500e18900f 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -38,7 +38,7 @@ requirements: - docopt {{ docopt }} - matplotlib {{ matplotlib }} - h5py {{ h5py }} - - cudatoolkit {{ cudatoolkit }} # [linux] + - cudatoolkit==9.0 # [linux] - tensorflow {{ tensorflow }} - pytorch {{ pytorch }} # [linux] - torchvision {{ torchvision }} # [linux]