ubuntu-Pytorch环境配置_ubuntu下pytorch环境的搭建

前言

          安装anaconda和安装驱动程序在之前的文章已经介绍过了，不清楚细节可以参考ubuntu-tensorflow环境配置_ubuntu tensorflow环境_呆呆珝的博客-CSDN博客，这里我就直接介绍pytorch环境的配置啦。

ubuntu-Pytorch环境配置

配置pytorch环境
(1)执行：conda create -n torch12  # torch12为环境名，可以自定义
         若提示找不到conda指令，执行 source ~/.bashrc

(2)执行：source activate torch12 # 激活环境，在终端开头会出现( torch12)的，证明环境已经激活

(3)  pip install torch==1.2
        pip install torchvision==0.4.0
        pip install opencv_python(自动安装最新版本，例如版本为4.5.1.48)
        pip install opencv_contrib_python==4.5.1.48

(4)执行：source deactivate torch12 # 退出环境

到此，pytorch训练环境已经搭建好了！！！

问题：如果cuda和cudnn不匹配，需要查询一下，在conda中安装啊！！！(后续可能出一篇针对cuda和cudnn的文章)