cuda 9.0 在ubuntu18下的虚拟环境中安装pytorch

1. source activate pytorch(该行代码是为了**conda虚拟环境,环境名为pytorch)如下图所示最前面若为pytorch表示已进入到该虚拟环境中。

cuda 9.0 在ubuntu18下的虚拟环境中安装pytorch

2.访问pytorch官网https://pytorch.org/

根据自己的电脑配置选择对应的版本,若没有v*n访问外网很慢,又不想修改conda的默认镜像的情况下可以考虑选用pip进行安装。

cuda 9.0 在ubuntu18下的虚拟环境中安装pytorch

3.在虚拟环境中执行上图中的run this Command等待完成即可。

4.验证是否安装成功

    输入python进入到python环境中,执行import torch命令,若未报错则表示pytorch安装成功,如下图所示:

cuda 9.0 在ubuntu18下的虚拟环境中安装pytorch

  输入torch.cuda.is_available命令,若出现如下信息则表示gpu版本的pytorch安装成功,否则你只能使用cpu版本的

cuda 9.0 在ubuntu18下的虚拟环境中安装pytorch

至此你便可以开心的训练你的模型拉!