pytorch1.4.0以及torchvision0.5.0的安装

我电脑原先的状况就是上面安装了原先的cuda10.0,然后你直接采用

conda install pytorch torchvision cudatoolkit=10.0

这个来安装已经不能满足需求了,因为装上的pytorch默认就是1.2.0的,然而跑我这个目标检测的代码要求pytorch版本>=1.4.0,所以在pytorch官网上找了半天,没有cuda10对应的windows版本的pytorch1.4.0,所以很尴尬,需要重新装cuda10.1(至少是这个版本,10.2应该也没问题),所以我最擅长的活又来了,我从学深度学习这半年以来在装环境上是真的话费了不少的时间,现在对这个流程都是驾轻就熟了,windows比Ubuntu更方便,安装cuda10.1的就请参照这个链接,这个上面安装的就是cuda10.0,你把10.0换成10.1就行,其他都一样。
装完cuda之后,你可以选择就在刚刚那个官网上把对应的.whl文件给下载下来,然后直接进入到下载目录,采用

pip install torch......

来进行安装,我就是这样装的,因为这么做可以看清自己安装的是哪个版本。你应该还可以直接去pytorch的官网上,用这个装:
pytorch1.4.0以及torchvision0.5.0的安装_第1张图片
他就会给你指定一个最新的pytorch和torchvision的版本,所以看你自己喜欢了。如果你用前一种方法的话,值得注意的一点是,你想要下载的那些文件,文件名中会注明cp35、cp36这样的标志,这就代表你要安装进的那个环境中的python版本(3.5、3.6)。
这样基本上就OK了,不过我在安装,程序出现了如下的报错:

from torchvision import _C
ImportError: DLL load faild.

这就证明你刚刚装的torchvision的版本不对,换一个版本就好了,我的就是将0.4.1换成0.5.0程序就正常了。
配环境真的是一种能力,它十分考验人的耐心,有些问题并不能像调试程序那样一步步去debug,只能按照网上的方法一步步去试,但当你环境装的多了,你的经验也丰富了,以后遇到问题可以节约更多的时间。如果遇到了很长时间都解决不了的问题,相信我,要冷静,总会有解决办法的,还是那句话,网上配环境的人那么多,肯定有很多仁兄前面遇到了和你类似的问题,他们中肯定有人对问题进行了总结,不要悲伤,不要心急,归纳问题,换一种搜索的方式也许就能解决问题。最后,祝你成功!

你可能感兴趣的:(深度学习及Python相关知识)