方法一：

pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

后面这个可以自己到官网上找自己对应的python和cuda版本的安装包，官网为链接：

方法二：

conda install pytorch torchvision cuda80 -c soumith

这里可以根据自己cuda版本确定。查看cuda版本，在python命令行里面输入：

import torch
torch.version.cuda

即可。

补充：关于win10环境下在使用torch调用cuda（）失败的原因

看了无数篇博客都是牛头不对马嘴，忙活了半天才搞清楚

如果大家在用torch的过程中看见这样的错误。。。。淡定，不是你的问题，也不是代码问题。。。是破win10的原因

只需要在代码前面加上

torch.cuda.current_device()
torch.cuda._initialized = True

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

猜你喜欢

• 使用python怎样写一个简单倒计时功能？

  怎样用Python写一个简单倒计时功能？对于倒计时功能的应用场景有很多，我们经常能在电商网站上看到，倒计时功能也是比较实用的，下面我们就来看看用Python怎样实现简单的倒计时。

• django与ajax怎样实现交互？一文带你看懂

  这篇文章给大家分享的是有关django与ajax怎样实现交互的内容，很多新手django与ajax交互可能不是很了解，因此分享给大家做个参考，有这方面学习需要的朋友接下来一起跟随小编看看吧。

• 用Python如何做堆栈和队列，原理和代码是什么

  堆栈是一个后进先出的数据结构，其工作方式就像一堆汽车排队进去一个死胡同里面，最先进去的一定是最后出来。我们可以设置一个类，用列表来存放栈中元素的信息，利用列表的append

• enumerate方法用于做什么，如何使用

  这篇文章主要介绍了Python中的 enumerate和zip，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下

• django如何连接数据库Mysql，详细操作是什么

  众所周知Django较为适合原生开发,即通过该框架搭建一个全新的项目,通过在修改models.py来创建新的数据库表,下面这篇文章主要给大家介绍了关于django连接Mysql中已有数据库的相关资料,需要的朋友可以参考下