学习PyTorch Day01

PyTorch 设计得更科学,无需像 TensorFlow 那样,要在各种 API 之间切换,操作更加便捷。
PyTorch 能够帮你快速实现模
型与算法的验证,快速完成深度学习模型部署,提供高并发服务,还可以轻松实现图像生
成、文本分析、情感分析等有趣的实验。
PyTorch 和tensorflow区别

PyTorch 会把我们需要计算的矩阵传入到 GPU(或 CPU)当中,在 GPU(或 CPU)中实
现各种我们所需的计算功能。因为 GPU 做矩阵运算比较快,所以在神经网络中的计算一般
都首选使用 GPU
使用 pip 安装 PyTorch
CPU 版本安装:
GPU 版本安装:(默认 CUDA 11.1 版本)

Linux

pip install torch1.9.0+cpu torchvision0.10.0+cpu torchaudio==0.9.0 -f http

Mac & Windows

pip install torch torchvision torchaudio

Linux & Windows

pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f

posted @ 2022-10-25 10:33  MIRyu  阅读(29)  评论(0编辑  收藏  举报