廣州網站建設怎么做福清市百度seo
1.什么是CUDA,什么是CUDNN
(1)什么是CUDA
CUDA(ComputeUnified Device Architecture),是顯卡廠商NVIDIA推出的運算平臺。 CUDA是一種由NVIDIA推出的通用并行計算架構,該架構使GPU能夠解決復雜的計算問題。
(2)什么是CUDNN
NVIDIA cuDNN是用于深度神經網絡的GPU加速庫。它強調性能、易用性和低內存開銷。NVIDIA cuDNN可以集成到更高級別的機器學習框架中,如谷歌的Tensorflow、加州大學伯克利分校的流行caffe軟件。簡單的插入式設計可以讓開發(fā)人員專注于設計和實現神經網絡模型,而不是簡單調整性能,同時還可以在GPU上實現高性能現代并行計算。
(3)CUDA與CUDNN的關系
CUDA看作是一個工作臺,上面配有很多工具,如錘子、螺絲刀等。cuDNN是基于CUDA的深度學習GPU加速庫,有了它才能在GPU上完成深度學習的計算。它就相當于工作的工具,比如它就是個扳手。但是CUDA這個工作臺買來的時候,并沒有送扳手。
想要在CUDA上運行深度神經網絡,就要安裝cuDNN,就像你想要擰個螺帽就要把扳手買回來。這樣才能使GPU進行深度神經網絡的工作,工作速度相較CPU快很多。
參閱該篇博客
2.在配置pytorch虛擬環(huán)境中遇到的問題
(1)如何查看電腦能夠支持的最高cuda版本
1)鼠標右鍵,點擊NVIDIA控制面板
2)打開左下角系統(tǒng)信息,切換到組件
(2)在pytorch的虛擬環(huán)境中,輸入 “torch.cuda.is_available()”,返回true,表示能夠使用gpu版的pytorch,但如果在運行程序的過程中出現"GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.",這表示顯卡的算力超出了pytorch虛擬環(huán)境中cuda支持的算力,此時需要提高cuda的版本
可參閱該篇博客
(3)在配置pytorch虛擬環(huán)境前如何選擇cuda以及pytorch版本
1)通過(1)可知,在虛擬環(huán)境中安裝的cuda版本不能大于此版本
2)通過(2)可知,虛擬環(huán)境中cuda版本支持的算力應大于等于顯卡的算力
不同cuda支持的計算能力情況
不同算力對應顯卡列表
3)通過1)、2)確定cuda版本后,根據NVIDIA驅動版本與CUDA版本的對應關系查看是否需要更新自己的顯卡驅動
NVIDIA驅動版本與CUDA版本對應關系
4)通過1)、2)確定cuda版本后,根據cuda與pytorch對應的關系確定pytorch版本
5)根據4)的版本在pytorch官網匹配pytorch的下載命令
PyTorch中torch、torchvision、torchaudio版本對應關系
3.在安裝gpu—pytorch中遇到的問題
(1)安裝gpu版的pytorch時能夠自動安裝cuda和cudnn嗎
通過命令行會自動將需要的cudatoolkit安裝好,不需要單獨去在網上搜索怎樣安裝cuda和cudnn
可參閱該篇博客
(2)電腦會自帶cuda嗎
一般來說顯卡正常工作的話會對應一個cuda版本,系統(tǒng)中的初始cuda也是別人裝的
可參閱該篇博客