赞
踩
很多年之前接触过UNet的细胞分割,这次突然拿出来做一个三维的腿骨分割,突然有点不知所措了,可能我的技术只局限于二维的吧,经过一个月的研究差不多鼓捣明白了一点,欢迎有经验的大佬赐教一些知识,我这就写写我自己做的过程吧。
一 训练前的准备工作
必要程序,环境就搞了我好多天,以为虚拟机就够了,结果配置还是不行,果断换成双系统,这个就不写了,网上太多了而且也不难,我以前都是用的虚拟机或者纯ubuntu系统没有装过,网上找了找一天也搞完了。
a. 操作系统:ubuntu18.04
Ubuntu 18.04.6 LTS (Bionic Beaver)
b. NVIDIA驱动:nvidia465.27(对应自己电脑配置在官网查看对应版本下载安装)
c. pytorch:1.8.0 Previous PyTorch Versions | PyTorch
CUDA:11.1
CUDNN:11.1
# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
d. Annaconda3:python3.6.9
这个在官网下载很慢,下面链接差不多可以达到1MB/s,还是不错的,找到对应的python版本即可,没有非要跟我一样,看你自己用哪个版本但是不要太高也不要太低。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。