Ubuntu18.04安装Pytorch
记录自己ubuntu18.04安装pytorch的心得,anaconda3-5.2已经安装好,这里不再详述。 目录 1.CPU版本Pytorch安装 2.GPU版本Pytorch安装 3.相关网址及总结 1.CPU版本Pytorch安装 CPU版本的pytoch安装比较简单,在 Pytorch官网 中选择python版本,运行conda或者pip命令即可,如下图所示。注意,为了方便多个Python版本的管理,通常需要在 anaconda中创建新的环境空间 。 2.GPU版本Pytorch安装 GPU版本的pytorch安装比较复杂,在安装pytorch之前,通常需要安装显卡驱动,cuda和cudnn, CUDA 是NVIDIA推出的用于自家GPU的 并行计算框架 ,也就是说CUDA只能在NVIDIA的GPU上运行; cuDNN (CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的 加速库 ,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型, cuDNN不是必须的,但是一般会采用这个加速库 。CUDA是必须的,cudnn是可选的。 通常需要考虑: (1)检查显卡型号是否支持GPU运算 检查方法:终端输入 ubuntu-drivers devices 得到以下结果: 图中,“model”对应的就是电脑显卡型号,