1、背景 2、安装支持cuda的pytorch 1.10.2 3、验证pytorch是否可用cuda 4、使用whl轮子加速安装
1、背景当下流行的深度学习框架pytorch,官网:
https://pytorch.org/get-started/locally/
其官网上有详细的安装教程,如下是Stable(1.10.2)
注意:使用支持cuda的pytorch,只需要安装nvidia显卡驱动程序即可!
Pytorch支持cuda的Stable(1.10.2),其pip安装命令如下:
pip3 install torch



