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

pytorch gpu安装 torch.cuda.is

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

pytorch gpu安装 torch.cuda.is

由于之前装了个假的gpu版本,所以来记录一下正确的怎么安装。

先看NVIDIA控制面板-帮助-系统信息-组件

看驱动版本,我的是9.0.

或者是  cmd输入NVIDIA-smi看 我的版本是385.54,然后看cuda toolkit对应版本,能装cuda9.0

一、CUDA Tollkit下载:https://developer.nvidia.com/cuda-toolkit-archive  

然后运行安装


二、对应版本的CUDNN下载:https://developer.nvidia.com/rdp/cudnn-download

然后重命名 为cudnn复制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0下面

然后配置环境变量,按照PyTorch(GPU版)详细安装教程_哔哩哔哩_bilibili

cuda装完之后cmd,nvcc -V验证一下

 

三、pytorch安装

创建虚拟环境 conda create -n  pytorch python=3.7

activate pytorch

然后是重点,pytorch下载从官网对应版本找命令容易出错。Previous PyTorch Versions | PyTorch找到页面最后能找到cuda9.0 对应的,下载后pip

进去找到对应的torch和torchvision

 然后进入到这两个文件所在的文件夹,E:   ,cd E:bishepytorch

pip install whl文件

最后进入python ,torch.cuda.is_available()是true,就成功了。

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

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

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