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()
看到类似下图说明安装成功。
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()
看到类似下图说明安装成功。
本文标题:Linux ssh安装PyTorch
本文链接:https://www.haomeiwen.com/subject/zikbictx.html
网友评论