当前位置:   article > 正文

安装gpu-torch(已经成功)_安装torchgpu

安装torchgpu

### !!!直接使用pip安装,不要使用conda安装,使用conda安装总是会安装成CPU版本。

1.第一次尝试

之前按照官网安装pytorch的命令进行安装,结果安装出来是cpu版本的,试过https://blog.csdn.net/u013468614/article/details/125910538里面的一些方法,但是使用conda安装出来还是CPU版本。

2.第二次尝试

然后,又尝试了把安装包下载到本地,使用
https://zhuanlan.zhihu.com/p/111430478这个方法,又出现了没有torch的错误,查找到说是可以按照pytorch官网的安装方法再安装一遍,但是不管是pip3还是conda,都会把原来的本地的gpu包进行升级,安装的torch也是CPU版本的,所以还是不行。在这里插入图片描述

3. 终于成功

最后,重新建了一个虚拟环境,使用这个一开始的方法,添加了清华镜像源之后直接使用官网pip命令安装,成功了

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
  • 1

在这里插入图片描述在这里插入图片描述
使用的教程是这个https://zhuanlan.zhihu.com/p/672526561

菜鸟折腾了好久终于好了,记录一下自己的操作过程,如果能对和我一样的朋友有帮助,那也是我的荣幸了。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/寸_铁/article/detail/975349
推荐阅读
相关标签
  

闽ICP备14008679号