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的文章)

你可能感兴趣的:(深度学习环境搭建,python,深度学习)