深度学习框架PyTorch安装
1、访问 https://pytorch.org/get-started/locally/
2、选择版本、操作系统、安装的平台、语言、计算平台。
3、cmd中执行:
使用国外源:用时超过24小时
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
使用国内源:用时10分钟左右
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/
4、安装完成。
5、查看PyTorch的版本信息。
import torch as t print(t.__version__)