pytorch居然提示我的GTX670性能太差,不能使用。 太坑了,之前不是用的好好的,也是比CPU跑的快。现在还想用咋办? 在换新的显卡之前只好凑合着用。到网上搜了一下,发现自己编译还能继续用。
先把代码clone 下来
git clone --recursive https://github.com/pytorch/pytorch cd pytorch
ubuntu安装cuda和cudnn非常方便,按照官方引导安装deb包即可使用。 cuda下载cudnn下载
USE_CUDA=True # 使用CUDA
最后安装
python3 setup.py install