利用GPU提升运算效率

首先由于在容器jupyter notebook中使用CPU运行速度过于慢,所以调用独显GPU进行计算,提高计算速度。

调用GPU大致流程是:

1.首先查看自己的电脑的情况:有NVIDIA GTX1650独立显卡,所以可以利用cuda来提升计算性能。

2.官网https://developer.nvidia.com/cuda-toolkit-archive下载cuda,并且在NVIDIA控制面板中修改参数。

3.安装CUDA,可以在cmd中输入 nvcc -V 查看是否安装成功。

4.配置NVIDIAKO控制面板。

如下图:

5.之后可以可以利用anoconda prompt环境运行py文件代码。可以检查任务管理器看是否成功调用GPU运行。