IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> Python知识库 -> PyCharm中安装GPU版本的PyTorch -> 正文阅读

[Python知识库]PyCharm中安装GPU版本的PyTorch

PyCharm中安装GPU版本的PyTorch

上个月,有个项目要用Pytorch跑,我就开始了安装之路。了解到Pytorch可以安装CPU和GPU版本的,想着GPU版本的更快,遂选择了安装GPU,但是CPU版本的与GPU安装类似,且安装起来更省时间,所以后面也介绍一下CPU安装。
网上很多资料都是利用Anaconda安装,博主不想再安装一个新的软件了(真的害怕环境报错,但是Anaconda确实比PyCharm集成了更多环境,感兴趣的可以试试。)

GPU版本的pytorch安装

第一步:判断是否可以安装GPU版本的pytorch

法一:

1.打开任务管理器
2.点击性能
3.点击左侧下拉框,往下拉,如果看到有GPU,则表示自己的电脑是有GPU的

法二:

计算机右击–>管理–>设备管理器–>显示适配器

法三(也能查看自己的适配的CUDA版本):

1.控制面板
2.在右上方的搜索框里输入NVIDIA
3.鼠标放在搜索出来的NVIDIA上
4.完成后,界面里可以查看NVIDIA的一些信息,显卡的驱动版本,如左侧菜单所示
5.点击帮助菜单,在下拉的菜单里选择系统信息选项,选择系统信息
6.在弹出的系统信息窗口里有两个菜单页面:显示和组件。选择组件,可以看到很多文件名
7.在3D设置模块,找到NVCUDA.DLL,在该行,可以看到该NVCUDA的版本

我这里已经装了CUDA10.2版本,所以显示的是10.2

第二步:安装CUDA

根据上面获得的版本号,选择对应的版本安装。注:一般不要安装太高的版本,而电脑一般也能兼容低版本的CUDA。
安装网址:https://developer.nvidia.com/cuda-downloads
安装其他版本的CUDA:https://developer.nvidia.com/cuda-toolkit-archive
这里以10.2版本为例进行安装步骤的介绍:

下载完成后,右键以管理员身份运行,选择合适的解压路径。

同意安装协议后,选择自定义安装(也可以选择精简安装):

单击下一步,取消勾选CUDA下面的VS(第一次安装也可以勾选,但是根据安装的软件有可能会报错,以防万一,取消。):

然后单击下一步,选择合适的安装位置:

点击下一步,等待安装成功。

测试安装是否成功:
在命令行中输入nvcc -V 查看版本号即可:

错误预警

我这里安装过程中出现了错误,翻阅众多博客后,大都推荐安装DDU,卸载本机驱动的方法,我试了之后再重新安装确实解决了。以下为解决网址:
解决地址1
辅助解决2(道理同1)

环境变量

我在安装的时候,系统自动配置了环境变量,所以并未出现错误。但我看很多博客中也有对环境变量的配置。具体可参考:
配置环境变量
上面的网址中配置环境变量的条目过多,我对比了其他博客和我个人的电脑,认为只有六条需要配置,有下图中的四条以及上面网址中Path中的两条。

第三步:安装PyTorch

我这里是在PyCharm中的终端中用pip命令下载的,也可以在命令行中进入相应的地址下载。具体步骤如下:
1.先新建一个项目或是打开需要装pytorch的项目
2.打开底部的terminal

3.进入pytorch官网,选择合适的参数配置,如pip安装、CUDA版本。

但是,本来该是安装命令的地方显示CUDA 10.2版本已不被支持。此时,我们应该选择上面的previous pytorch versions,查看之前的版本。

在其中选择合适的CUDA版本后,复制命令到pycharm中的terminal中执行即可。但是直接这样下载的话速度较慢,这里我采用了清华镜像,复制以下命令执行:

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

下面问题又来了,报错:

这里如果搜黑体字、红体字、黄体字都会出来一系列的解决文章,但是我认为黄体字提示pip版本需要升级的问题不用管,之前改了反而报错了(心累)。黑体字提示证书有问题,红体字提示版本的问题,我也没有找到一个较好的解决方案。后来根据清华源下载的问题,我从黑体字入手解决的,就是把他添加为信任,直接在代码后面加上信任的指令就好了。

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.htm
l --trusted-host pypi.tuna.tsinghua.edu.cn

参考解决文章:
解决网址1
解决网址2
安装成功:

测试是否安装成功以及是否能使用CUDA:

import torch

print(torch.__version__)
# 1.6.0

print(torch.cuda.is_available())
# True

补充(参考别的博客):
cuda安装成功print(torch.cuda.is_available())却输出为false的原因有二:
1.版本号不匹配
2.pytorch不小心安装为cpu版,与cuda适配的pytorch应该为gpu版本。
?
?
?
至此,全部安装完成。

  Python知识库 最新文章
Python中String模块
【Python】 14-CVS文件操作
python的panda库读写文件
使用Nordic的nrf52840实现蓝牙DFU过程
【Python学习记录】numpy数组用法整理
Python学习笔记
python字符串和列表
python如何从txt文件中解析出有效的数据
Python编程从入门到实践自学/3.1-3.2
python变量
上一篇文章      下一篇文章      查看所有文章
加:2022-09-04 01:07:23  更:2022-09-04 01:09:55 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年12日历 -2024/12/26 13:30:52-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码
数据统计