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

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