Windows 10 python3.6安装pytorch

  近年来,深度学习框架如雨后春笋般的涌现出来,如TensorFlow、caffe、caffe2、PyTorch、Keras、Theano、Torch等,对于从事计算机视觉/机器学习/图像处理方面的研究者或者教育者提高了更高的要求。其中Pytorch是Torch的升级版,其有非常优秀的前端和灵活性,相比TensorFlow不用重复造轮子,易于Debug调试,极大的提高开发效率,使得其在其他框架中脱颖而出。更多信息参见:caffe2+Pytorch1.0 = Pytorch1.0,期待即将推出的1.0版本,caffe2更适合后端部署。

  1.进入pytorch官网https://pytorch.org/

选择对应的版本、环境、安装方式等:

将生成的命令在终端进行输入,安装成功。

测试:

import torch#导入没有报错说明安装成功

print(torch.cuda.is_available())#查看是否支持GPU
原文地址:https://www.cnblogs.com/gcter/p/10766827.html