转着圈救国:如何配置Jetbrains IDEA/PyCharm远程调用Windows上的Python解释器并使用CUDA加速

2019-06-20 分类 [Python开发]

手头的电脑都只有CPU,最高级的就是个GTX650Ti,兴冲冲的装上了个GPU加速的Tensorflow,直接报错告知算力不足。于是就看上了楼下的8700K+1066,虽然也不怎么样,但是总比老i5吭哧吭哧强多了。但是由于项目需要,只能装了Windows 10,因此想像往常一样IDEA/PyCharm中添加Remote Interpreter是不行了。毕竟人家是Windows,开了WSL可以解决SSH的问题,但是WSL本身不支持CUDA,因此只能“充分利用”Core i7 8700的算力,总有些隔靴搔痒之感。某天突发奇想,WSL是可以直接运行Windows的exe的,因此通过WSL远程Windows上的Python是否可以呢?如果可以那CUDA是否也能正常调用呢?于是开始一番倒腾,因为过于“婉转”,曲线救国已经不足以描述其曲折程度,故题为“转着圈救国”。

首先从远程SSH登陆到Windows 10的WSL中,这里已经通过远程桌面在WSL中启动了SSHD服务,试着定位到Windows的Python安装路径下,激动地键入python.exe,成功:

在WSL的SSH上运行python.exe

良好的开始就是成功了一半嘛,没毛病,下一步安装CUDA:

安装CUDA

安装成功,下一步安装PyTorch CUDA版本:

(未完待续)