本文共 956 字,大约阅读时间需要 3 分钟。
前两天无意看到一篇论文《EISEN: A PYTHON PACKAGE FOR SOLID DEEP LEARNING》,介绍了一个基于Pytorch的医学图像处理库,正好和所研究的内容有关,既然有人已经造好了轮子,那不是很好吗,于是找到和开始安装,将遇到的坑记录一下:
我当前的系统状况如下:硬件Windows10+2080Ti,cuda10.0,cudnn7.3,已经安装了pytorch1.1-GPU,TensorFlow1.13-GPU和Keras-2.2.4-GPU,PaddlePaddle0-GPU。看了github上该库需要pytorch1.4支持,上Pytorch官网去看,1.4版本需要cuda10.1支持,看了网上有些说需要卸载之前装的cuda10.0,然后再去安装cuda10.1,由于我这些虚环境里面已经装了很多库,担心更换后导致用不了,所以采用了以下方法顺利安装:
conda install --use-local pytorch-1.4.0-py3.6_cuda101_cudnn7_0.tarconda install --use-local cudatoolkit-10.1.243-h74a9793_0.tar
这样只是在虚环境中变动了cuda版本,没有影响到其他虚环境。安装完毕后不会影响其他虚环境。
在不同的虚环境中试了一下,一切OK,可以放心了。
两个cuda版本和谐共存,Conda环境管理真的是好用。
pip install --upgrade eisen
接下来就是好好研究这个库如何用在CT、X光片上了
转载地址:http://zyqrb.baihongyu.com/