栏目分类:
子分类:
返回
文库吧用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
文库吧 > IT > 软件开发 > 后端开发 > Python

安装pytorch 后torch.cuda.is

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

安装pytorch 后torch.cuda.is

今年安装pytorch 后一直只能安装上CPU版本,不能使用GPU。

参考了这篇博客的方法:完美解决torch.cuda.is_available()一直返回False的玄学方法(完美解决torch.cuda.is_available()一直返回False的玄学方法_python_脚本之家 (jb51.net))

因为我添加了清华源镜像,所以使用conda下载会直接选择清华源下载,后来上清华源的网站上看了下,可能是清华源没有对应cuda版本的pytorch,所以会装cpu版的。

上清华源网站查看安装的版本(我安装的是cuda 11.3,安装的pytroch版本为1.12.1)

 后面发现我在conda的虚拟环境安装的为python 3.8版本

重新删除环境后

conda remove -n pytorch --all

重新新建环境并安装python版本为3.10:

conda create -n pytorch python=3.10

安装完成后,在按照pytorch官网的安装

最后测试torch.cuda.is_available() 返回True,大功告成! 

 

转载请注明:文章转载自 www.wk8.com.cn
本文地址:https://www.wk8.com.cn/it/1038095.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 wk8.com.cn

ICP备案号:晋ICP备2021003244-6号