ubuntu16+pytorch+cuda.is_available安装排坑
前情说明:nvidia-smi中的cuda版本和 nvcc -V查询的不一致转载:https://blog.csdn.net/ljp1919/article/details/102640512总结:cuda分为运行(run)版本和驱动(driver)版本“pytorch安装版本” 依赖于 “cudatoolkit安装版本” 依赖于 “显卡的驱动版本”(1) 显卡驱动版本nvidia-smi # 查询显卡驱动型号查询结果如下,驱动版本号:430.26(2) 匹配对应的 cudatool.
原创
2020-06-17 09:50:20 ·
330 阅读 ·
0 评论