zzzhhh:

Since PyTorch is nothing different from these CUDA samples in terms of calling CUDA interface, if all CUDA APIs PyTorch is calling are those in samples that I can compile and execute, PyTorch should be good to run.


I wouldn’t claim that’s the case, as e.g. an NVIDIA driver update without a restart would leave the machine in an “undefined behavior” status to my experience. I.e. some applications might work, other might crash. However, based on the samples it seems that at least the toolkit is partially working.



 zzzhhh:

Is it possible I install a PyTorch in my user space (or build up one from source) and this PyTorch version can see the GPU?


Yes, you could build PyTorch from source, but I would still claim that your setup might be in a flaky state.