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

2021-10-05

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

2021-10-05

首先自己先安装好anoconda,网上随便找个教程就好!

安装/检查显卡驱动
win+R启动cmd,可以在cmd里输入nvidia-smi命令查看当前驱动版本号。
正常显示如下:

462.30是此电脑当前驱动的版本号,CUDA 11.2是当前这个驱动能适配的最高CUDA版本。观察Driver Version的值是否大于400,如果小于请更新显卡驱动。

要使用pytorch-GPU,首先确保自己的显卡是英伟达显卡(RTX),然后才可以安装CUDA。

进入Anaconda prompt下操作

conda create -n 环境名 python==版本
conda create –n pytorch python==3.

安装成功后输入activate pytorch
可以成功进入到pytorch环境。
输入pip list
看到当前环境下安装的文件,可以看到并没有pytorch,所以需要安装。

安装pytorch
进入官网 pytorch官网
根据自己的显卡配置来选择下载文件,选择相应的操作系统,如果电脑无N卡,那就选择CPU。

复制代码到命令行:conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

接下来就是慢慢等待安装啦!

输入pip list

可以看到有pytorch的存在。

检测安装
命令行输入python进入python编译器环境。

输入import torch,如果输入后没有任何报错,那就是成功了。

再输入torch.cuda.is_available(),如果返回True,则安装到此完成。

接下来就是好好享受你的深度学习之旅啦!

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

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

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