I installed it using official conda install
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

the install was pretty normal (I will try to get the installation log but maybe it’s not very informative).
I also tried official pytorch/pytorch:latest docker image but still torch could find cuda version but cuda available was false.
The host machine I’m using doesn’t have nvidia-docker or support nvidia runtime. The gpu devices and libcuda* were manually mounted (if i ran nvidia-smi the docker was able to find gpus, driver version, and cuda version).
Are there ENV variables that pytorch search for in order to bind to cuda devices?