The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37

lilyntalmers: <p>I’m trying to install some packages in Conda and run a python program. But when I try to run the program it throws this error:</p>
<pre><code>UserWarning:
NVIDIA GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3070 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
</code></pre>
<p>I’m installing all the packages using these commands:</p>
<pre><code>conda create -y -n styleflow python==3.6.7
conda activate styleflow
conda install -y git
git clone https://github.com/Tuba202/StyleFlow-Made-Easy.git StyleFlow
cd StyleFlow/
conda install -y pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.2 -c pytorch
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html
conda install -y tensorflow==1.14 tensorflow-gpu==1.14
pip install torchdiffeq==0.1.0 tensorflow==1.14 tensorflow-gpu==1.14 scikit-image scikit-learn requests qdarkstyle qdarkgraystyle pyqt5 opencv-python
</code></pre>
<p><strong>I’m trying to use this repo <a href="https://github.com/Tuba202/StyleFlow-Made-Easy" rel="noopener nofollow ugc">StyleFlow</a>.</strong></p>
<p><em>I tried to find solutions but nothing seems to work. Installing PyTorch and Tensorflow is the hardest thing I’ve ever tried.</em></p>

ptrblck: <p>In your install command you are explicitly specifying an old PyTorch release with the CUDA 10.2 runtime, which is too old for your Ampere GPU. Use any install command posted on our website and it would work since the default CUDA runtime for the current PyTorch release is 11.7.<br>
I also don’t understand why you are trying to install multiple (old) PyTorch binaries from conda and pip in the same environment.</p>
<blockquote>
<p>Installing PyTorch and Tensorflow is the hardest thing I’ve ever tried.</p>
</blockquote>
<p>It is as easy as <code>pip install torch</code> and I’m unsure how you came up with your install command using old releases.</p>

Marc_Joshua: <p>Hi, I meet the same problem. Because i want use fcos3d module built on mmdetection3d framework, when i use latest pytorch version, error reporting " THC/THC.h: No such file or directory". And i found this “<a href="https://github.com/CoinCheung/pytorch-loss/pull/37" class="inline-onebox" rel="noopener nofollow ugc">remove THC.h by zyddnys · Pull Request #37 · CoinCheung/pytorch-loss · GitHub</a>” tells the file has been removed after torch1.11. And i have been struggling for a long time.</p>

ptrblck: <p>Yes, the TH(C) namespace is deprecated and this header file was removed.<br>
I see <a href="https://github.com/open-mmlab/mmdetection3d/issues/1332">this issue</a> raised the same error, but it also seems that <code>mmdet3d/ops/ball_query/src/ball_query.cpp</code> was removed and I didn’t see any <code>includes</code> for <code>TH(C)</code> anymore. Which file is raising the error in your build?</p>

Marc_Joshua: <p>i have solved this problem by reinstall all env dependencies, thanks.</p>

king2: <p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/2/2/221802309e178d9f9dc0d37799a596924797b10c.png" data-download-href="https://discuss.pytorch.org/uploads/default/221802309e178d9f9dc0d37799a596924797b10c" title="Pyt1"><img src="https://discuss.pytorch.org/uploads/default/original/3X/2/2/221802309e178d9f9dc0d37799a596924797b10c.png" alt="Pyt1" data-base62-sha1="4RBFFuQrF2rpnhbHZ3p3T3rDJBa" width="690" height="265" data-dominant-color="222222"><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">Pyt1</span><span class="informations">1109×426 47.7 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>I face the same problem. As your explanation, we should consider the current cuda arc whille installing PyTorch and that is understandable. But the project developed with old PyTorch refuse to built on the new environment. The figure above is the sample I tried to execute. It works with CPU.</p>

ptrblck: <aside class="quote no-group" data-username="king2" data-post="6" data-topic="167239">
<div class="title">
<div class="quote-controls"></div>
<img loading="lazy" alt="" width="24" height="24" src="https://discuss.pytorch.org/letter_avatar_proxy/v4/letter/k/c68b51/48.png" class="avatar"> king2:</div>
<blockquote>
<p>But the project developed with old PyTorch refuse to built on the new environment.</p>
</blockquote>
</aside>
<p>Could you explain this issue in more detail, please?</p>

king2: <p>The following two errors happened with the higher cuda version (11.1)<br>
/media/…/models/dla/DCNv2/src/cuda/dcn_v2_cuda.cu(126): error: identifier “THCudaBlas_SgemmBatched” is undefined</p>
<p>/media/…/models/dla/DCNv2/src/cuda/dcn_v2_cuda.cu(273): error: identifier “THCudaBlas_Sgemm” is undefined</p>
<p><del>RuntimeError</del><del>:</del> <del>Error </del><del>compiling </del><del>objects </del><del>for </del><del>extension</del></p>

ptrblck: <p>The THC namespace was migrated to ATen, e.g. in <a href="https://github.com/pytorch/pytorch/pull/42553">this PR</a>, so you would need to replace all legacy calls with their corresponding <code>ATen</code> ones. I’m not familiar with the repository you are using and thus also don’t know how much work it would be.</p>