当前推荐的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版本