正常用 conda 安装的 cuda-toolkit(pytorch-cuda) 是runtime,有一些包是依赖 dev 的,这时需要物理主机安装 nvidia-cuda-toolkit,要查看物理主机是否安装了 cuda-toolkit:
nvcc --version
如果不想物理主机安装 cuda-toolkit,还可以使用conda 安装CUDA Compiler:
conda install nvidia/label/cuda-12.1.0::cuda-nvcc
这样就可以正常安装需要编译的包了,又不会改变物理主机的环境。