在windows上编译pytorch-1.3.1的那些坑
由于用的计算卡比较老,官方下载的pytorch gpu版本不支持显卡,在使用gpu部分时显示
no kernel image is available for execution on the device
就起了自己编译pytorch包的念头,中间遇到几个坑,记录一下
0,安装nvidia 驱动,cudatool,anaconda, vs2017就不说了,网上到处有
1,github下载...
原创
2019-12-31 10:05:52 ·
2347 阅读 ·
0 评论