环境配置小知识点

1、torch-gpu:
一、windows下conda/pip install 安装的torch 和 trochvision 是没经过GPU编译的,也就是说是用不了GPU,要去https://download.pytorch.org/whl/torch_stable.html下载编译过的torch,torchvision,但是这又会出现新的问题,
二、直接安装wheel,from torch._C import * ImportError: DLL load failed: 找不到指定的模块。
这是numpy等文件依赖的原因,解决:conda install numpy pyyaml mkl cmake cffi,切记如果添加了清华源也要再后面 -i, 不然pip还是连接超时。
三、安装torch 和 torchvision版本一定要和cudn,cudnn版本对应。不要安装显卡驱动版本,这库安装跟那个木的关系。
四、好了,到这一步你的windows上面的torch-gpu可以用了,这时候你一定很高兴,可惜。。。windows自身的原因,它和下面的keras一样,gpu利用率不高!!!回头是岸吧,这时候只有一个办法,在linux上面训练。hhhhhhhhhhh…苦涩滴笑容
2、keras:
需要tensorflow-gpu,但是好像安装的时候,第二列会有个gpu,这个就可以!!即tensorflow gpu,但是即使使用了gpu,它在windows上面的利用率也不高.

你可能感兴趣的:(环境配置小知识点)