Linux ssh安装PyTorch

2019-11-18  本文已影响0人  卖点儿什么

https://pytorch.org/get-started/locally/

v/tensorflow 镜像(python3.5.2    cuda10.0.130)

pip3 install torch torchvision

pip3 install torch torchvision

测试安装成功没

python3

import torch

torch.cuda.current_device() 

看到类似下图说明安装成功。

上一篇 下一篇

猜你喜欢

热点阅读