Hi, I am trying to run a script from github https://github.com/vsitzmann/deepvoxels 4. I have a RTX 2060 GPU on my laptop and a RTX 2080Ti GPU on a server. I use the exact same conda environment on both machines, both have CUDA version 10.1. I am able to run the code without a problem on my laptop, however I can’t make it work on the server.
Initially, the code didn’t work on either machine. I simply updated the version of pytorch on my laptop and it solved the problem. I ran this command: conda install pytorch torchvision cudatoolkit=10.0 -c pytorch on the server , it updated pytorch and torchvision as asked however the same problem remains. Here is the ouput
Begin training...
Traceback (most recent call last):
  File "run_deepvoxels.py", line 395, in <module>
    main()
  File "run_deepvoxels.py", line 386, in main
    train()
  File "run_deepvoxels.py", line 183, in train
    grid2world=grid_origin)
  File "/store/usagers/tamez/deepvoxels/projection.py", line 90, in comp_lifting_idcs
    voxel_bounds_min, voxel_bounds_max, _ = self.compute_frustum_bounds(camera_to_world, world2grid)
  File "/store/usagers/tamez/deepvoxels/projection.py", line 73, in compute_frustum_bounds
    p = torch.bmm(camera_to_world.repeat(8, 1, 1), corner_points)
RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450

This is my first post here so if I did not provide some information please let me know.
Thanks in advance.