配置CUDA+GPU环境
写得很好的博客:
Win10+MX250+CUDA10.1+cuDNN+Pytorch1.4安装+测试全过程(吐血)
https://blog.youkuaiyun.com/weixin_40313940/article/details/104152125
更正错误:
- 如果电脑没有VStudio需要勾选“对号”
- 判断是否能够使用GPU加速,返回True,则说明安装成功
torch.cuda.is_available()
Win10+MX250+CUDA10.1+cuDNN+Pytorch1.4安装+测试全过程(吐血)
https://blog.youkuaiyun.com/weixin_40313940/article/details/104152125
更正错误:
torch.cuda.is_available()