Linux查看cuda和pytorch版本,解决版本不匹配问题
写这篇文章主要是方便记录Linux一些常见的命令。1.查看cuda版本:nvcc -V或者cat /usr/local/cuda/version.txt2. 查看pytorch版本:pythonimport torchprint(torch.__version__)3. 查看pytorch和cuda版本是否匹配:print(torch.cuda.is_available())返回True说明版本是匹配的。4.解决pytorch和cuda版本不匹配问题:
原创
2021-11-30 11:30:23 ·
4395 阅读 ·
0 评论