My environment is:
PyTorch version: 1.7.1
Is debug build: False
CUDA used to build PyTorch: 11.0
Python version: 3.9.2
Is CUDA available: True
CUDA runtime version: 9.1.85
Versions of relevant libraries:
[pip3] numpy==1.21.0
[pip3] pytorch-lightning==1.1.6
[pip3] torch==1.7.1
[pip3] torch-lr-finder==0.2.1
[pip3] torch-tb-profiler==0.3.1
[pip3] torchaudio==0.7.0a0+a853dff
[pip3] torchmetrics==0.3.2
[pip3] torchvision==0.8.2
[conda] blas 1.0 mkl anaconda
[conda] cudatoolkit 11.0.3 h15472ef_8 conda-forge
[conda] mkl 2020.2 256 anaconda
[conda] numpy 1.21.0 pypi_0 pypi
[conda] pytorch 1.7.1 py3.9_cuda11.0.221_cudnn8.0.5_0 pytorch
[conda] pytorch-lightning 1.1.6 pypi_0 pypi
[conda] torch-lr-finder 0.2.1 pypi_0 pypi
[conda] torch-tb-profiler 0.3.1 pypi_0 pypi
[conda] torchaudio 0.7.2 py39 pytorch
[conda] torchmetrics 0.3.2 pypi_0 pypi
[conda] torchvision 0.8.2 py39_cu110 pytorch
onnx: ‘1.10.2’
onnxruntime: ‘1.10.0’