I cannot build the environment with the code provided:
conda env create -f environment_cuda11.yml
Collecting package metadata (repodata.json): done
Solving environment: failed
ResolvePackageNotFound:
- pytorch==1.7.1=py3.8_cuda11.0.221_cudnn8.0.5_0
I also tried to build the environment manually, but torch-lightning is not compatible with pytorch 1.7.1. Even old torch-lightning like 1.4.0, 1.5.0 is still not compatible. It will automatically uninstall my old torch 1.7.1 and install the new 2.1.0 pytorch when I install torch-lightning.
Please consider update the code to support new versions of pytorch.
I cannot build the environment with the code provided:
conda env create -f environment_cuda11.yml
Collecting package metadata (repodata.json): done
Solving environment: failed
ResolvePackageNotFound:
I also tried to build the environment manually, but torch-lightning is not compatible with pytorch 1.7.1. Even old torch-lightning like 1.4.0, 1.5.0 is still not compatible. It will automatically uninstall my old torch 1.7.1 and install the new 2.1.0 pytorch when I install torch-lightning.
Please consider update the code to support new versions of pytorch.