安装pytorch和pyTorchvision

pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/

此命令在安装torchvision的同时,会把torch一起安装。

import torch
import torchvision

安装结束后重新开启一个终端并进入python环境,输入以下命令,如果没报异常,则说明安装成功了。

查看pytorch版本号
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/yxyou_1124/article/details/127972031