Python+PyCharm+PyTorch+Cuda/GPU 安装步骤
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤
- 一、 无nvidia显卡,直接用CPU跑程序时
- 二、有Nvidia显卡,想用GPU跑程序时
- 2.1 先查版本!
- 2.2 下载安装
- 2.3 验证是否安装成功
gadflycq 2023-2-14
一、 无nvidia显卡,直接用CPU跑程序时
1.1 python官网python.org下载最新版本python,安装python
1.2 pycharm官网https://www.jetbrains.com/pycharm下载community版,安装pycharm
1.3 直接在cmd命令行中输入指令:“pip3 install torch torchvision torchaudio”完成安装pytorch
二、有Nvidia显卡,想用GPU跑程序时
2.1 先查版本!
(1) 确定显卡支持的cuda版本。
在cmd命令行中输入命令:“nvidia-smi”回车。(如提示命令不可用,可先执行命令:cd C:\Program Files\NVIDIA Corporation\NVSMI)
找到驱动号,例如:385.54
(2) 根据驱动版本号,查找对应的cuda版本号
(3) 根据cuda版本号,确定pytorch版本号
(4) 根据pytorch版本号,确定python版本号。(新版本可直接在官网看到下载选项,下表主要是旧版本对应关系)
Pytorch | Python |
---|---|
pytorch-1.5 | Python 3.7 |
pytorch-1.4 | Python 3.6 |
pytorch-1.3 | Python 3.6 |
pytorch-1.2 | Python 3.6 |
pytorch-1.1 | Python 3.6 |
pytorch-1.0 | Python 3.6 |
pytorch-0.4 | Python 3.6 |
pytorch-0.3 | Python 3.6 |
pytorch-0.2 | Python 3.5 |
pytorch-0.1 | Python 3 |
2.2 下载安装
(1) 根据上面查到的版本号,先在python官网python.org下载对应版本python,安装python
(2) pycharm官网https://www.jetbrains.com/pycharm下载community版,安装pycharm
(3) 在Nvidia官网下载对应cuda版本驱动并安装(不确定这一步是不是必须的,没验证过)
最新版:https://developer.nvidia.com/cuda-downloads
旧版本:https://developer.nvidia.com/cuda-toolkit-archive
(4) 在cmd命令行中使用“pip3 install torchvision”,将自动从官网下载安装dataclasses、numpy、pillow、torch、torchvision、typing-extensions一系列工具包。(此时自动下载安装的是CPU版本的torch)
(5) 参照下面的地址,下载对应版本cuda和pytorch的whl文件到本地,然后直接在cmd命令行中,用“pip3 install 本地路径文件名”的方式安装。(此步骤安装完后,将会用本地包含cuda的torch包替换掉上面第4步自动安装的torch)
1、pytorch1.1.0 (1)CPU版本
https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
(2)GPU版本
https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
(CUDA 10.0)
https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
(CUDA 9.0)2、版本pytorch1.0.1 (1)CPU版本
https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
(2)GPU版本
https://download.pytorch.org/whl/cu80/torch-1.0.1-cp36-cp36m-win_amd64.whl
(CUDA 8.0)
https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl
(CUDA 9.0)
https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl
(CUDA 10.0)3、版本pytorch1.0.0 (1)CPU版本
https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
(2)GPU版本
https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl
(CUDA 8.0)
https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl
(CUDA 9.0)
https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-win_amd64.whl
(CUDA 10.0)4、版本pytorch0.4.1 (1)CPU版本
https://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
(2)GPU版本
https://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
(CUDA 8.0)
https://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
(CUDA 9.0)
https://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
(CUDA 9.2)
其他各版本的pytorch以及torchvision,的CPU版本,GPU版本都可以在下面的网址下载:
https://download.pytorch.org/whl/torch_stable.html
以及pytorch官网给出的旧版本在线命令方式安装:
https://pytorch.org/get-started/previous-versions/
2.3 验证是否安装成功
在pycharm或python IDE里面运行以下代码:
import torch
if torch.cuda.is_available(): #返回True则说明已经安装了cudaprint('GPU: ' + torch.cuda.get_device_name(0))
else:print('No cuda')from torch.backends import cudnn #判断是否安装了cuDNN
print('cuDNN:' + str(cudnn.is_available()))
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤相关推荐
- Anaconda下pytorch的详细安装步骤
Anaconda下pytorch的详细安装步骤 文章目录 Anaconda下pytorch的详细安装步骤 准备工作 1.安装Anaconda 2.Cuda 安装pytorch 1.Anaconda下新 ...
- python3.7安装步骤-Windows下Python 3.7.0的安装步骤,Python370
Windows下Python 3.7.0的安装步骤 由于Python版本太新的话,能会出现不稳定的情况,所以在这里,作者建议用版本不太新的Python.在这里,我用的是Python3.7.0版本(其他 ...
- python+pycharm+selenium自动化测试环境搭建步骤
python+pycharm+selenium自动化测试环境搭建步骤 原文链接:https://blog.csdn.net/weixin_51429941/article/details/116850 ...
- Python的初步认知与安装步骤 (小白必备)
Python的初步认知与安装步骤 Python初步认知 安装步骤 基础代码测试(变量.数据类型) 运算符.条件结构.循环结构 思维导图 Python初步认知 Python简介 Python是一种跨平台 ...
- Anaconda Pycharm Pytorch(GPU版本)的配置
写在前面的注意事项! 记得在每次搞新项目的时候先配置一个pytorch环境,或者复制已有的环境 conda create -n 新环境名 --clone 旧环境名 之后可以 conda list一下( ...
- PyTorch\CUDA软件安装指引(Windows)
目录 [前言] [快速安装指引] [Pytorch先决条件关系图] [操作系统&VS Build Tools] [Annaconda--Python虚拟环境创建] [显卡驱动更新&GP ...
- PyTorch+CUDA+cudnn安装
1.安装CUDA 先查看一下电脑显卡对应的CUDA驱动版本 上图可以看到CUDA驱动版本为11.1.114,那就下载11.1版本的CUDA好了.(这里应该也不是必须对应,只是说有可能下载低版本的会有问 ...
- Pytorch CPU/GPU 安装方法。
首先,查看自己有没有显卡及其显卡驱动.没有显卡,就CPU安装就行. windows版,CUDA,cudnn,和Anaconda安装方法这里不详述.参考一下连接 2022最详细,最新的 Win11/WI ...
- python 3.7.732位安装步骤_Python 3.7.7详细图文安装教程(附安装包) | 我爱分享网
软件下载: Python 3.7.7_64位正式版:[ 安装前须知: 1.Python3.7.7适用于Win7/8/10(32/64位)系统,亲测可用! Python3.7.7 WIN10 64位安装 ...
最新文章
- 2015年第11本:代码整洁之道Clean Code
- python难学嘛-终于明白0基础学python难吗
- 华为网赛云数据中心基础原理自测答案
- 监狱中的物联网用例:从尘土飞扬的监狱到智能监狱设施
- Linux--生产者与消费者
- linux——常用指令
- 手机html文件转TXT,Html 转换 Txt
- 基于.NET实现数据挖掘--聚类分析算法
- python线程池如何调度,python线程池控制
- android apk更新下载以及安装
- Apple Pay 详解
- 我理解的myisam引擎之六 MYI、MYD文件的解析
- 游戏辅助小助手框架构想
- 我国互联网遭境外网络攻击
- ts 在vue中的使用总结
- 和老婆的一次真实对话
- 简易的HTML网页设计
- 如何像程序员一样思考
- js 匿名函数和闭包
- 海南大专计算机应用技术,2022年海南单招计算机应用技术专业公办学校名单