PyTorch CPU 和 GPU 版本环境搭建及 PyTorch 基本语法
2023-04-18 14:06:36 时间
1、PyTorch CPU 版本环境搭建
PyTorch 官网:https://pytorch.org/ 在首页中部选择要安装的 PyTorch 配置,当前计算平台选择 CPU 版本。
在本机命令行输入:
pip3 install torch torchvision torchaudio
注意,一定要是pip3
,pip
是不能安装成功的。
安装完毕,查看 PyTorch 版本,表明安装成功。
2、PyTorch GPU 版本环境搭建
PyTorch 官网:https://pytorch.org/ 在首页中部选择要安装的 PyTorch 配置,当前计算平台选择 CUDA 11.3 版本。
在命令行输入:
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
注意,一定要是pip3
,pip
是不能安装成功的。
安装完毕,查看 PyTorch 版本,表明安装成功。
>>> python
>>> import torch
>>> torch.__version__
'1.10.0'
>>> exit()
3、PyTorch 基本语法
HUB 模块: PyTorch 官网地址:https://pytorch.org/hub/ GitHub 地址:https://github.com/pytorch/hub
相关文章
- 六个优秀的可解释AI (XAI)的Python框架推荐
- Python虚拟环境很简单,看完你就会了
- 总结几个简单好用的Python人脸识别算法
- 又是好兄弟 爱立信与三星和解专利纠纷
- Flink分布式程序的异常处理
- 好用的 Python 虚拟环境,没有之一
- 为什么TensorFlow可以做机器学习开发?
- 记一次beego通过go get命令后找不到bee.exe的坑
- 指定Task任务顺序执行
- CompletionService 使用小结
- Win环境安装Protobuf 2.0 版本
- shell 同时执行多任务下载视频
- Seata 分布式事务解决方案及特点
- zookeeper 使用api 进行节点增删改查及实现简易的配置中心
- 潘石屹首次Python考试成绩 99 分,失分点:画完图后忘了隐藏画笔的箭头
- 啃完Python基础,你要这样做才行
- Python开发者调查显示只有十分之一的人还在用Python 2
- TIOBE 5 月编程语言排行榜:C语言跑到第一,python连续两年上涨
- Python 是前浪,Julia 是后浪?
- 不朽 C++ 为新贵 Python 应用提速 8000 倍!