tensorflow-gpu&pytorch(GPU) 安裝指南

想要用gpu加速得先安裝CUDA和cuDNN。 NVIDIA的顯卡驅動程序和CUDA完全是兩個不同的概念哦!CUDA是NVIDIA推出的用於自家GPU的並行計算框架,也就是說CUDA只能在NVIDIA的GPU上運行,而且只有當要解決的計算問題是可以大量並行計算的時候才能發揮CUDA的作用。 CUDA的本質是一個工具包(ToolKit);但是二者雖然不一樣的。 1.到官網查找版本關係 pytorch
相關文章
相關標籤/搜索