cpu跑代码慢到怀疑人生,还得是gpu啊
但是你得有gpu,还得安装cuda。。。。。。一系列操作之后才能跑
1、首先查看笔记本cuda版本,安装对应cuda。在英伟达控制面板查看即可(网上很多,自己查)
安装教程在这cuda安装
cuda向下兼容,所以安装比笔记本对应版本低的也可以,但高的不行
2、安装cuda版本的pytorch包
首先得查看torch、python的版本,在这里下载对应版本的包python库
3、安装
在安装包的路径下:
pip install 下载的包的名字
注意:先安装torch,再安装torchvision
全部完成之后,运行代码,自动使用gpu:
![在这里插入图片描述](https://img-blog.csdnimg.cn/d97db99f73934745a12b154416a9b7ca.png)