大多數人用的電腦的顯卡配置可能沒有那么強,也就是說,我們很難享受到最新版本pytorch給我們帶來的模型訓練的速度和效率,為此,我們需要想辦法在現有顯卡情況下部署應用pytorch。
筆者有一臺電腦,顯卡算力很低,那么以該電腦為例,為大家介紹如何部署應用pytorch功能。
1、電腦基本情況
本人電腦的顯卡是一臺GeForce 730顯卡,如下圖所示:
前期,本人曾一頓鼓搗,也安裝了cuda,版本為11.4。
使用nvcc -V命令,也可以看到安裝的CUDA工具包的版本
在python安裝了pytorch的虛擬環境中,用torch.cuda.is_available()命令來檢查PyTorch是否能夠使用GPU也是正常的。
torch.cuda.get_dev