【Centos7】CentOS 7下的PyTorch安装策略:高效实践指南
大家好 我是寸铁👊
总结了一篇【Centos7】CentOS 7下的PyTorch安装策略:高效实践指南✨
喜欢的小伙伴可以点点关注 💝
前言
由于需要跑深度学习,要用到
pytorch
,那Centos7
怎么安装Pytorch
? 跟着寸铁走不迷路!
安装
Step1
-
安装的
CUDA
版本取决于你的显卡信息 -
根据
CUDA
版本在官网中查看安装的命令
- 如果需要更早前的版本,则点击左下角
Step2
输入安装的命令:
如我这里是:
pip3 install torch torchvision torchaudio
- 1
这里需要确保你的
pip3
可以正常使用
等待一段时间即可