ubuntu系統+CUDA9.0+pytorchGPU+訓練CNN

運行環境及安裝

前言

將不定長字符轉換爲訂場字符的識別問題,並使用CNN完成訓練和驗證:

  1. 賽題數據讀取(封裝爲Pytorch的DataLoder)
  2. 構建CNN模型(使用Pytorch)
  3. 模型訓練與驗證
  4. 模型結果預測

環境安裝
1.現在Anaconda中創建一個專門用於本次練習的虛擬環境。

conda create -n py37_torch131 python=3.7

檢查更新當前conda

conda update conda

2.激活環境並安裝pytorch1.3.1

3.通過命令一鍵安裝所需其它依賴庫

pip install jupyter tqdm opencv-python matplotlib pandas

4.啓動notebook

問題
發現步驟3一直不成功,出現一下問題:

可能因素:
檢查上面的步驟,發現是上一個未安裝成功,如圖

需要換源
解決方案:
應用conda裏的pip換源,安裝一個pqi的包進行換源,命令如下:
pip install pqi
1
看源

pqi ls
1
換源

pqi use aliyun
1
解決結果:
完成,超級快

其他操作
安裝Py包
#安裝 virtualenv
conda install virtualenv
1
2
#查看 已經安裝的包列表
conda list
1
2
#查看 指定環境下的py包
conda list -n py名字
1
2
刪除虛擬環境
conda remove -n your_env_name(虛擬環境名稱) --al
1
刪除環境中的某個包
conda remove --name $your_env_name $package_name
1
關閉虛擬環境(即從當前環境退出返回使用PATH環境中的默認python版本)
#Windows下:
deactivate
1
2
#Linux/Mac下:
source deactivate
1
2
修改python的path
vim ~/.bash_profile
1
conda的虛擬環境的位置

類似

/opt/anaconda3/envs/py27_MuXueNet/bin/python

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章