pytorch related

环境相关

# 检查环境是否都没有冲突或者污染,按照官网默认安装torch的cuda版本可能与系统的不一致
torch.version.cuda
# cuda version 和 nvidia driver version 有依赖关系,尤其是装torch >= 1.13 
# https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
cat /proc/driver/nvidia/version  
cat /usr/local/cuda/version.txt 

你可能感兴趣的:(深度学习)