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 安装步骤相关推荐

  1. Anaconda下pytorch的详细安装步骤

    Anaconda下pytorch的详细安装步骤 文章目录 Anaconda下pytorch的详细安装步骤 准备工作 1.安装Anaconda 2.Cuda 安装pytorch 1.Anaconda下新 ...

  2. python3.7安装步骤-Windows下Python 3.7.0的安装步骤,Python370

    Windows下Python 3.7.0的安装步骤 由于Python版本太新的话,能会出现不稳定的情况,所以在这里,作者建议用版本不太新的Python.在这里,我用的是Python3.7.0版本(其他 ...

  3. python+pycharm+selenium自动化测试环境搭建步骤

    python+pycharm+selenium自动化测试环境搭建步骤 原文链接:https://blog.csdn.net/weixin_51429941/article/details/116850 ...

  4. Python的初步认知与安装步骤 (小白必备)

    Python的初步认知与安装步骤 Python初步认知 安装步骤 基础代码测试(变量.数据类型) 运算符.条件结构.循环结构 思维导图 Python初步认知 Python简介 Python是一种跨平台 ...

  5. Anaconda Pycharm Pytorch(GPU版本)的配置

    写在前面的注意事项! 记得在每次搞新项目的时候先配置一个pytorch环境,或者复制已有的环境 conda create -n 新环境名 --clone 旧环境名 之后可以 conda list一下( ...

  6. PyTorch\CUDA软件安装指引(Windows)

    目录 [前言] [快速安装指引] [Pytorch先决条件关系图] [操作系统&VS Build Tools] [Annaconda--Python虚拟环境创建] [显卡驱动更新&GP ...

  7. PyTorch+CUDA+cudnn安装

    1.安装CUDA 先查看一下电脑显卡对应的CUDA驱动版本 上图可以看到CUDA驱动版本为11.1.114,那就下载11.1版本的CUDA好了.(这里应该也不是必须对应,只是说有可能下载低版本的会有问 ...

  8. Pytorch CPU/GPU 安装方法。

    首先,查看自己有没有显卡及其显卡驱动.没有显卡,就CPU安装就行. windows版,CUDA,cudnn,和Anaconda安装方法这里不详述.参考一下连接 2022最详细,最新的 Win11/WI ...

  9. 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位安装 ...

最新文章

  1. 2015年第11本:代码整洁之道Clean Code
  2. python难学嘛-终于明白0基础学python难吗
  3. 华为网赛云数据中心基础原理自测答案
  4. 监狱中的物联网用例:从尘土飞扬的监狱到智能监狱设施
  5. Linux--生产者与消费者
  6. linux——常用指令
  7. 手机html文件转TXT,Html 转换 Txt
  8. 基于.NET实现数据挖掘--聚类分析算法
  9. python线程池如何调度,python线程池控制
  10. android apk更新下载以及安装
  11. Apple Pay 详解
  12. 我理解的myisam引擎之六 MYI、MYD文件的解析
  13. 游戏辅助小助手框架构想
  14. 我国互联网遭境外网络攻击
  15. ts 在vue中的使用总结
  16. 和老婆的一次真实对话
  17. 简易的HTML网页设计
  18. 如何像程序员一样思考
  19. js 匿名函数和闭包
  20. 海南大专计算机应用技术,2022年海南单招计算机应用技术专业公办学校名单

热门文章

  1. 理解paxos算法--事前咨询,事中决定,事后提交
  2. PCA9685--16路 PWM模块舵机驱动板--STM32 IIC接口模块
  3. ZigBee网络类毕业论文文献都有哪些?
  4. 80%的经理人都不知道的邮件常识
  5. Android 安卓动画 属性动画 - 缩放动画
  6. 安卓动画开始,暂停和停止
  7. 用Python调用Graphviz生成复杂股权关系图
  8. taro 小程序编译在标签上px转rpx的api
  9. python爬虫精进参考答案_高校邦数据科学创新通识课【Python爬虫】课后习题答案...
  10. 关于电子科技大学宿舍安装热水的调查