簡介
pytorch是非常流行的深度學習框架。下面是Windows平台配置pytorch的過程。
一共需要安裝cuda、pycharm、anancoda、pytorch。
主要介紹cuda和pytorch的安裝。
安裝cuda
1. 根據自己的顯卡，選擇合適的cuda版本。
百度輸入CUDA，進入官網下載。
下載結束后，進行安裝。
安裝結束后，自動彈出此窗口。
一路下一步
在環境變量中，添加cuda的路徑
在cmd中驗證是否安裝成功
可顯示出CUDA版本，說明安裝已成功。
2. 下載pytoch
登陸www.pytorch.org進行選擇自己電腦的配置。
將下方的命令，復制到cmd中執行，注意需用管理員權限打開cmd。
驗證是否安裝成功
從cmd中，進入python環境
import torch
torch.__version__
torch.cuda.is_available()
出現以下輸出torch版本和'true'，則說明安裝成功。