python pytorch使用gpu加速
2021/6/27 11:14:13
本文主要是介绍python pytorch使用gpu加速,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
一、查看GPU驱动
nvidia-smi
根据cuda版本确定去官网查看torch版本,运行查询出来的代码
https://pytorch.org/get-started/locally/
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
这篇关于python pytorch使用gpu加速的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-08有遇到过吗?同样的规则 Excel 中 比Python 结果大
- 2024-03-30开始python成长之路
- 2024-03-29python optparse
- 2024-03-29python map 函数
- 2024-03-20invalid format specifier python
- 2024-03-18pool.map python
- 2024-03-18threads in python
- 2024-03-14python Ai 应用开发基础训练,字符串,字典,文件
- 2024-03-13id3 algorithm python
- 2024-03-13sum array elements python