1.首先需要安装对应的显卡驱动
这里我对应的显卡的安装所在地址如下
显卡安装地址
如果你的显卡不同的话,还需要修改其他的步骤
2.接着需要下载对应的CUDA软件
CUDA是显卡厂商NVIDIA推出的运算平台,是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。开发者可以使用C语言来为CUDA架构编写程序。
所以tensorflow以及pytorch下面写的包相当于调用的CUDA的库,然后运行对应的程序。
显卡装入主机之后编写程序所需的安装步骤
最新推荐文章于 2023-08-15 07:59:29 发布