要尝试使用paddle还有tensorflow,需要CUDA。对不起,因为太小白我还问了问客服,联想拯救者r9000x,确认有显卡驱动支持CUDA但是CUDA需要自己安装。
我们需要安装CUDA、、
安装:
找合适的CUDA型号:
直接开始菜单找到NVIDIA控制面板:
cuda官网安装,很不幸,11.2没有win11,用了10
卸载了framview sdk:
安装时遇到
“You already have a newer version of the NVIDIA Frameview SDK installed”,先把电脑已经存在的FrameView SDK 卸载掉,把C:\Program Files\NVIDIA Corporation\FrameViewSDK文件夹删掉
记录一下第一个安装的存储地点
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.2
查看环境变量,我的有了,不需要自己添加了,这是有的证据
(没有的话需要手动加入路径哈,在这,比照这我那个路径加一下,其他大佬有教程,yyds
下载cudnn
我下载了cuda11.2,以为11.x代表了所有的11.几,结果发现不行,那个代表了11.6!于是下载了
解压,文件夹内的东西复制到CUDA对应的文件夹过去,也就是上面我们记录的地址(我的是默认安装,或许我们会一样):
to:
这样子粘贴:
验证cuda是否安装成功:
CMD:
cd CUDA安装目录\extras\demo_suite,
运行:
bandwidthTest.exe
deviceQuery.exe
得到PASS,成功:
电脑的 GPU 能否被 PyTorch 调用:
我们部署torch到最后可以直接调用GPU一共要下三个东西:
- CUDA(√)
- CUDNN(√)
- torch(这个里面有cpu以及GPU版本)
见下一章