Linux ssh安装PyTorch
2019-11-18 本文已影响0人
卖点儿什么
https://pytorch.org/get-started/locally/
v/tensorflow 镜像(python3.5.2 cuda10.0.130)
pip3 install torch torchvisionpip3 install torch torchvision
测试安装成功没
python3
import torch
torch.cuda.current_device()
看到类似下图说明安装成功。