Use `pytorch`
@@ -34,12 +34,12 @@ requirements:
- pip
- swig
- openmm
- - pytorch-gpu # [cuda_compiler_version not in (undefined, 'None')]
+ - pytorch # [cuda_compiler_version not in (undefined, 'None')]
- ocl-icd # [linux]
- khronos-opencl-icd-loader # [osx]
run:
- python
- - {{ pin_compatible('pytorch-gpu') }} # [cuda_compiler_version not in (undefined, 'None')]
+ - {{ pin_compatible('pytorch') }} # [cuda_compiler_version not in (undefined, 'None')]
- {{ pin_compatible('openmm', min_pin='x.x', max_pin='x.x') }}
- ocl-icd # [linux]
- ocl-icd-system # [linux]
