博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Windows下深度学习环境CUDA10.1和CUDA10.0共存
阅读量:2491 次
发布时间:2019-05-11

本文共 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,由于我这些虚环境里面已经装了很多库,担心更换后导致用不了,所以采用了以下方法顺利安装:

  • 在NVIDIA官网下载显卡的最新驱动并安装,貌似不更新驱动后面的步骤不行。
  • 建立conda虚环境,在虚环境中使用`conda install pytorch torchvision cudatoolkit=10.1 -c pytorch`安装Pytorch1.4-GPU,不出意外的话会因为网络太慢中断,但会提示下载pytorch-1.4.0-py3.6_cuda101_cudnn7_0.tarcudatoolkit-10.1.243-h74a9793_0.tar的链接,在IDM中复制链接下载很快,本地安装用如下命令
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环境管理真的是好用。

  • 安装eisen,参照官网指令即可
pip install --upgrade eisen

接下来就是好好研究这个库如何用在CT、X光片上了

转载地址:http://zyqrb.baihongyu.com/

你可能感兴趣的文章
css3 动画的播放、暂停和重新开始
查看>>
IOS 上传ipa文件失败
查看>>
eclipse Android 开发基础 Activity 窗体 界面
查看>>
怎样玩转千万级别的数据
查看>>
input输入框修改后自动跳到最后一个字符
查看>>
Windows与Linux之间海量文件的传输与Linux下大小写敏感问题
查看>>
HDU 3948 不同回文子串个数
查看>>
Leetcode: Valid Parentheses
查看>>
Python
查看>>
自己动手开发调试器 01
查看>>
Python基础-包
查看>>
POJ 2696 计算表达式的值
查看>>
都江堰很美-佩服古人_Crmhf的一天
查看>>
Linux系统资源查询命令(cpu、io、mem)
查看>>
(转)PowerHA完全手册(一,二,三)
查看>>
hdu 4391 Paint The Wall 线段树 +优化 2012 Multi-University Training Contest 10 )
查看>>
socket与socketServer通信
查看>>
Android_Layout (一)
查看>>
《过早退出是一切失败的根源》读后感
查看>>
luogu P1774 最接近神的人_NOI导刊2010提高(02)
查看>>