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,并且换源。教程有很多,搜关键字就好啦。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值