擅长:python、mysql、java
<p>通常可以在命令行中执行此操作:</p>
<pre><code>nvcc version
</code></pre>
<p>您可能必须运行:</p>
<pre><code>sudo apt install nvidia-cuda-toolkit
</code></pre>
<p>也是</p>
<hr/>
<p>正如另一位回答者所提到的,您可以:</p>
<pre><code>torch.cuda.nccl.version()
</code></pre>
<p>在皮托克。将此复制粘贴到您的终端:</p>
<pre><code>python -c "import torch;print(torch.cuda.nccl.version())"
</code></pre>
<p>我相信tensorflow中也有类似的东西</p>