cuda版本

当前推荐的CUDA版本是10.2和11.3,这两种版本支持大多数的PyTorch版本。您可以通过以下方法查看您的CUDA版本:

  • 在Windows上,打开命令提示符(CMD),输入nvcc --version

  • 在Linux上,您可以查看/usr/local/cuda/version.txt文件,或者运行cat /usr/local/cuda/version.txt命令。

请确保安装的CUDA版本与您的显卡兼容,并且满足您所使用的深度学习框架(如PyTorch)的要求。如果您需要安装特定版本的CUDA,可以使用conda或pip命令进行安装,例如:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

请根据您的环境选择合适的CUDA版本

Top