cuda11.0+cudnn+pytorch1.7.1+mmsegmentaion安装指南(步骤简单)

  1. 安装cuda11.0:在官网进行.run文件的下载
  2. 安装cudnn:下载安装包文件
  3. 安装pytorch:去官网按照命令进行安装(建议使用conda命令)
  4. 安装mmcv(必须先安装mmcv,再安装mmseg):我安装了1.4.4的版本
  5. 安装mmsegmentation:如果是普通的安装,就按照官网的提示安装就好了。如果是开发者模式,也就是说你之后要改框架里面的内容,写你自己方法的代码,就使用git clone 把mmsegmentation的master的代码全都下载下来,然后 cd mmsegmentation, pip install -e .(注意空格!!)

一些避坑:

  1. 能不能在同一个主机下面安装不同版本的cuda?可以的,因为cuda的调用终究是一个调用目录的东西,详细操作见链接:https://www.zhihu.com/collection/758974054。
  2. 安装python包的时候不小心安装到base环境下面了?首先,必须激活你自己的conda 环境。其次,使用pip命令的时候,可能调用的不是当前环境的pip命令,而是base 环境的pip 命令。所以建议使用pip 命令的时候,进入自己环境的python目录下面,运行 python -m pip install XXX。(可能有点绕哈哈哈哈)所以!!!使用conda命令进行安装可以避免这个问题。尤其是安装torch的时候,官网给出了conda和pip两种安装方式,选择conda会好很多。
  3. 安装pytorch网速太慢:建议开VPN,并且换源。教程有很多,搜关键字就好啦。

你可能感兴趣的:(语义分割,经验分享,均值算法,python)