您当前的位置: 首页 >  ar

柳鲲鹏

暂无认证

  • 0浏览

    0关注

    4642博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument

柳鲲鹏 发布时间:2019-03-26 19:38:39 ,浏览量:0

原因是显卡用的RTX 2080Ti,CUDA就要装10以上,这个时候,请看发生了变化:

 

https://pytorch.org/resources

页面最下,通过选择可以看到:

pip install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp27-cp27m-linux_x86_64.whl

 

关注
打赏
1665724893
查看更多评论
立即登录/注册

微信扫码登录

0.0564s