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.