/home/system/miniconda3/envs/main/lib/python3.8/site-packages/torch/cuda/__init__.py:230: UserWarning: NVIDIA GeForce RTX 5070 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation. The current PyTorch install supports CU