安装gpu版torch

在官网下载的gpu版本torch总是cpu版

因为这不是真的gpu版!!!

搞了好久简直要疯掉555555

解决方案

通过本地whl文件的方式安装。

来到该网站:https://download.pytorch.org/whl/torch

可以看到各个版本的torch,xuanze符合自己版本的就行
torch-1.13.0+cu116-cp37-cp37m-win_amd64.whl,意为torch1.13.0、cuda11.6、python3.7、win 64位系统的版本。

下载到本地后安装就可以

pip3 install D:\torch-1.13.0+cu116-cp37-cp37m-win_amd64.whl 

你可能感兴趣的:(pytorch,深度学习,人工智能)