PyTorch – 安装和入门

安装PyTorch

官网的安装方式 https://pytorch.org/get-started/locally/
从pypi.org安装(GPU版本)
https://pypi.org/project/torch/#files
https://pypi.org/project/torchvision/#files
选择自己的python和cuda版本,比如python3.5就选择torch-1.2.0-cp35-cp35m-manylinux1_x86_64.whl
cuda9.0选择兼容版本torch==0.4.1
使用 torch.__version__ 查看版本号

测试GPU是否可用

PyTorch使用指定GPU

1.在终端中指定

2.使用Python代码

3.使用函数set_device

张量的使用

参考资料

官方网站 https://pytorch.org/
中文文档 https://pytorch-cn.readthedocs.io/zh/latest/
官网教程中文版 http://pytorch123.com/
kaggle教程 https://www.kaggle.com/kanncaa1/pytorch-tutorial-for-deep-learning-lovers

PyTorch – 安装和入门》有1个想法

发表评论

电子邮件地址不会被公开。 必填项已用*标注