1.背景

學(xué)習(xí)深度學(xué)習(xí)的話,肯定需要安裝PyTorch和TensorFlow,安裝這兩個深度學(xué)習(xí)框架之前得安裝CUDA.

CUDA是什么?

CUDA是一個并行計算平臺和編程模型，能夠使得使用GPU進(jìn)行通用計算變得簡單和優(yōu)雅。Nvidia官方提供的CUDA 庫是一個完整的工具安裝包，其中提供了 Nvidia驅(qū)動程序、開發(fā) CUDA 程序相關(guān)的開發(fā)工具包等可供安裝的選項。

2.安裝過程

(1)選擇安裝版本:

首先打開自己電腦上的NVIDIA控制面板,點擊系統(tǒng)信息

點擊"組件"

這里顯示的就是本機的版本,我這里是11.1.大部分電腦都可以向下兼容,所以也可以安裝版本小于11.1的CUDA

(2) 到CUDA官網(wǎng),點擊這里

https://developer.nvidia.com/cuda-downloads

這里我選擇的是10.1版本的CUDA

(3)根據(jù)自己情況進(jìn)行選擇,然后點擊下載

(4)下載完成時,以管理員身份運行

(5)點擊OK

(6)等待安裝

(7)等待ing,兼容性檢查

(8)點擊同意

(9)選擇自定義

(10)點擊下一步

(11)新建文件夾,類似這種

(12)然后選擇安裝位置,前兩個選擇CUDA1,最后一個選擇CUDA2,然后點擊下一步

(13)勾選之后,然后點擊Next

(14)等待安裝

(15)點擊下一步

(16)點擊關(guān)閉

(17)檢驗是否安裝成功

win + R,輸入cmd打開Windows終端

輸入