Unable to install pytorch from source for version 2.0.0

kernelpanic77: <p>I am trying to build pytorch from source to enable support for MPI in pytorch DDP. I have cloned the source code and switched to the v2.0.0 branch, after which I have updated the submodules.<br>
GCC version: 8.3.0<br>
cuda version is 11.0<br>
I am using intel MPI in my system.<br>
The installation is in a conda environment for a HPC cluster.</p>
<p>I have also attached screenshots of nvidia-smi.<br>
<img src="https://discuss.pytorch.org/uploads/default/original/3X/a/7/a7f6f3aa43f451e9bda3f11b3d3b7c3bae19c4b8.png" alt="Screenshot from 2024-03-12 23-20-54" data-base62-sha1="nXSPqbpJ1xZNJDaSReVzPZ0R5fa" width="648" height="434"><br>
<img src="https://discuss.pytorch.org/uploads/default/original/3X/0/b/0b82f879d80870f5604fc1656b2a00c4117410ce.png" alt="Screenshot from 2024-03-12 23-12-40" data-base62-sha1="1DPR2comSU9FYX97dXd90YETK5o" width="632" height="110"><br>
<img src="https://discuss.pytorch.org/uploads/default/original/3X/0/0/008eafc4e4ef96e7444d58fcafd0db1fa5a47b30.png" alt="Screenshot from 2024-03-12 23-12-25" data-base62-sha1="4VHGAf0BlsKPXlZEG9heMFe3x6" width="611" height="100"><br>
<img src="https://discuss.pytorch.org/uploads/default/original/3X/5/a/5ae33d12b1f5db273373305296c93e47cd1a7eb6.png" alt="Screenshot from 2024-03-12 23-11-54" data-base62-sha1="cY1PnC5z3bJdjYKzwhIYgfMZpS6" width="456" height="60"></p>
<p>Upon executing all steps as mentioned in the readme, I executed <code>python setup.py develop</code> after which I get the following errors.</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/8/f/8f27dab16adf1331e6ad5f72c17533121ef75938.png" data-download-href="https://discuss.pytorch.org/uploads/default/8f27dab16adf1331e6ad5f72c17533121ef75938" title="Screenshot from 2024-03-12 23-25-52"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/8/f/8f27dab16adf1331e6ad5f72c17533121ef75938_2_690x366.png" alt="Screenshot from 2024-03-12 23-25-52" data-base62-sha1="kqpFgLNefRvuw3OkncOOMFDiPfa" width="690" height="366" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/8/f/8f27dab16adf1331e6ad5f72c17533121ef75938_2_690x366.png, https://discuss.pytorch.org/uploads/default/optimized/3X/8/f/8f27dab16adf1331e6ad5f72c17533121ef75938_2_1035x549.png 1.5x, https://discuss.pytorch.org/uploads/default/optimized/3X/8/f/8f27dab16adf1331e6ad5f72c17533121ef75938_2_1380x732.png 2x" data-dominant-color="202429"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">Screenshot from 2024-03-12 23-25-52</span><span class="informations">1422×756 178 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p>/scratch/ishans/pytorchgpu/pytorch/aten/src/ATen/cuda/CublasHandlePool.cpp:112:24: error: ‘cublasSetWorkspace’ was not declared in this scope<br>
TORCH_CUDABLAS_CHECK(cublasSetWorkspace(handle, workspace_it-&gt;second.get(), getChosenWorkspaceSize()));<br>
^~~~~~~~~~~~~~~~~~</p>
<p>I am unable to debug this issue. Any idea on how I can fix this ?</p>