安装pytorch0.4.0

Posted 2021-01-02 www-caiyin-com

参考了官网https://pytorch.org/previous-versions/中的说明

(jj1env) [[email protected] down_python0.4.0]$ pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl

这样就pytorch就安装完了。

pip install torchvision 

这样就安装了 torchvision  ，当然你也可以安装torchvision 的指定版本，如 pip install torchvision==0.1.8

 

然后就验证下是不是安装成功了。于是我就

进入python环境。然后，继续输入

import torch

import torchvision

如果没有报错，就是安装成功了。