For 2, the memory I mean the CPU memory. If user only run import tensor, it should not load any CUDA related libraries into CPU memory per my understanding. I compared the shared libraries of this 2 version. For cuda11.1 one, the libtorch_cuda_cu.so and libtroch_cuda_cpp.so are new and both are very big, not sure whether it is related or not. @albanD , any idea?
For torch with cuda11.1:

image1427×716 54.4 KB

For torch with cuda10.2

image1453×654 52.1 KB