ubuntu18.04下安装pytorch-gpu版

前面 之前介绍过TensorFlow-gpu版本的安装过程。现在再说说pytorch的安装过程。 1.安装Ubuntu后,先安装nvida显卡驱动。 2.安装anaconda。 3.安装cuda。 4.安装cudnn 5.安装pytorch-gpu 前面5个步骤 参考 TensorFlow的安装过程中。 现在主要介绍下pytorch-gpu的安装。 5、安装pytorch-gpu 进入官网 选择对
相关文章
相关标签/搜索