pytorch安裝詳細步驟

環境配置步驟

如果顯卡支持GPU，在命令行處，輸入nvidia-smi ，查看自己驅動的版本。
參考文章

官網選擇型號

官網

清華源快速安裝

參考博文

安裝完檢驗

輸入torch.cuda.is_available()，顯示True則意味着可以使用GPU。

pycharm設置項目環境

這一步是容易忽略導致操作錯誤的地方，設置環境項目注意如果是導入已有的虛擬環境，需要選擇conda environment，如果是新建虛擬環境，則選擇virtualenv environment。
這裏因爲需要導入已有的安裝pytorch的環境，所以正確操作如下:

之後可以看到對應的編譯器已經添加到：

項目顯示編譯器未設置，進行再次設置。