环境:Ubuntu20.04.1 首先,安装英伟达驱动,安装英伟达驱动之前需要禁用Ubuntu自带的第三方驱动nouveau,否则无法运行英伟达的驱动,安装英伟达的驱动可参考这个链接:https://zhuanlan.zhihu.com/p/59618999,装完驱动用Nvidia-smi看一下是否有cuda,如果有的话直接装对应的pytorch就行了,需要注意的是,pytorch仅支持cuda9.2、10.1、10.2和11.0.
Ubuntu安装英伟达驱动和cuda以及pytorch-gpu版本
最新推荐文章于 2024-07-05 10:37:48 发布