1.conda activate env(进入虚拟环境)
2.python(进入python)
3.import torch(导入torch包)
4.print(torch.version)(查看torch版本)
5.print(torch.version.cuda)(查看torch对应的cuda版本)
6.conda deactivate(退出虚拟环境)
nvcc -V/nvcc --verion/nvidia-smi (查看环境cuda版本)
1.conda activate env(进入虚拟环境)
2.python(进入python)
3.import torch(导入torch包)
4.print(torch.version)(查看torch版本)
5.print(torch.version.cuda)(查看torch对应的cuda版本)
6.conda deactivate(退出虚拟环境)
nvcc -V/nvcc --verion/nvidia-smi (查看环境cuda版本)