Pytorch 的安裝

GPU版本的安裝

• Windows平台 CPU 版本安裝

conda install pytorch torchvision cpuonly -c puython

　　Windows平台需安裝VC，需要的聯系博主。

• Linux平台 CPU 版本安裝

conda install pytorch torchvision cpuonly -c pytorch

• MacOS平台 CPU 版本安裝

conda install pytorch torchvision -c pytorch

GPU版本的安裝

• GPU版本需有 NVIDIA 顯卡硬件支持，請確保您的顯卡支持 cuda，具體型號可查詢如下網址：https://developer.nvidia.com/cuda-gpus
• 檢查顯卡驅動

　　在命令行或者shell中執行：

 nvidia-smi     

　　執行完會打印出您的顯卡信息：

　　　　　　

　　在右上角，顯示您的顯卡目前的cuda驅動版本，比如上圖顯示為11.4，要注意，這里並不是表明您已經安裝了cudatookit，僅僅代表，您當前驅動可以支持的 cuda 最高版本。

• 通過pytorch官網選擇安裝命令，打開pytorch官網（https://pytorch.org/），在網頁中間位置，有安裝提示欄，如下所示：

　　　　　　

 　　這里本人電腦采用如下命令安裝：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

• 檢驗Pytorch安裝完成：

　　命令行進入 python 使用：

>>> import torch >>> torch.randn(4,5)

　　若生成結果，安裝完成。

• 驗證GPU安裝

>>> torch.cuda.is_available()

　　返回 True 則安裝完成。