1. 程式人生 > 實用技巧 >究極pytorch環境搭建

究極pytorch環境搭建

不使用GPU

(用於比較小的專案,稍微大點的都要用GPU,如果硬體允許最好裝GPU版)

1

下載Anaconda

2

開啟Anaconda Navigator - Enviroment, 它自帶一個base環境
在這裡插入圖片描述

左下角Creat一個環境, 這個環境裡有python的版本選擇,所以不用單獨下一個python
在這裡插入圖片描述
這裡我創了一個pytorch環境,用來專門跑pytorch框架的專案

3

開啟剛剛創的環境, 點綠色三角形, open Terminal
在這裡插入圖片描述

這裡就進入了剛剛搭的虛擬環境中, 進入pythorch官網 點此進入

4

在這裡插入圖片描述
CUDA選擇None就是CPU版 不使用GPU,把Command語句複製到剛剛的框裡, 回車

然後y 等他安裝。
(使用Conda裝,會把所有的包都給你裝進去,很大 建議用Pip裝,只會裝torch框架的東西,快很多,一樣的複製語句回車)在這裡插入圖片描述
安裝完成之後在這環境中輸入 python ,>>> impor torch 如果沒報錯,說明成功了, 也可以使用pip list看一看裡面有沒有torch

怎麼看是不是在這個環境中:
在這裡插入圖片描述
前面括號內是什麼, 就是在什麼環境中, 可以從Anaconda Navigator-三角-open Terminal開啟這個環境中的框

5

我使用的是pycharm
在pycharm裡File- setting-Project:XXXX這是你建立專案檔案的名字(torch框架只能在這個資料夾中作用)

在這裡插入圖片描述
點選右上角設定- Add…
在這裡插入圖片描述
點Existing enviroment 右邊… 選擇python.exe的路徑, 選擇你剛剛Anaconda Nevigator中建立那個環境資料夾裡的那個python.exe(在 Anaconda安裝的檔案加里/envs/你建立環境的名字, 這裡面的那個python.exe )
這一步就是把Anaconda中建立的環境附在pycharm上
Apply一下, 等包載入完, 就可以了。


太晚了,GPU版明天在弄


10.13:
RuntimeError: CUDA error: no kernel image is available for execution on the device

conda install pytorch1.0.0 torchvision0.2.1 cuda80 -c pytorch

GeForce GT 730 with CUDA capability sm_35 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the GeForce GT 730 GPU with PyTorch, please check the instructions at
cuda pytorch 下載版本過高,顯示卡算力跟不上
想找改setting中顯示卡算力的方法找不到
現在解決方法: anaconda創個新環境。 裝python3.7+cuda8.0
!!!!!!成功