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

在pytorch中使用GPU

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

在pytorch中使用GPU

今天学习的时候突然发现我的torch.cuda.is_available()返回的是False。这不能忍,我可是高贵的3060用户啊。得改。首先查看我使用的是11.4版本的CUDA

然后查看我的pytorch版本

原来我装的是cpu版本的torch。那去官网看看我应该下载哪个版本的。Previous PyTorch Versions | PyTorch

看了看之前版本最高支持到11.1的,那看看最新版本的。torch-1.10,能支持CUDA 11.3的,就决定是你了。去吧conda,使用

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 

 遇到了一些问题。

 复制粘贴到搜索引擎,在 installation - PackagesNotFoundError: The following packages are not available from current channels: - Stack Overflow提示我使用把conda-forge加入channels。然后就可以了。

conda config --append channels conda-forge

了解一下原因,答主说channels是存放packages的服务器,而从社区维护的conda-forge启动在标准的channels不可用时是一个很好的方法。

如果一些包在windows上不被支持,那可能就只能用pip安装了。这样还是会安装在你的conda目录下,但是是使用python内置的包管理工具。有时候有些包只能用pip下载。

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

版权所有 (c)2021-2022 MSHXW.COM

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