This copies the existing PyTorch conda example and modifies the existing PyTorch 1.5 version to avoid incompatible GPUs. It adds a second example that uses CUDA 11.1, which is compatible with newer GPUs such as the A100.
If this works, we can add similar requirements to our other examples that use CUDA 10.x.
This requires some testing to confirm it works on the expected GPUs and is not run on unexpected GPUs.
This copies the existing PyTorch conda example and modifies the existing PyTorch 1.5 version to avoid incompatible GPUs. It adds a second example that uses CUDA 11.1, which is compatible with newer GPUs such as the A100.
If this works, we can add similar requirements to our other examples that use CUDA 10.x.
This requires some testing to confirm it works on the expected GPUs and is not run on unexpected GPUs.