Windows 下 PyTorch 入门深度学习环境安装与配置 GPU 版

1.确定自己的硬件信息,确定电脑有英伟达 (NVIDIA)显卡

        在任务栏上右键打开任务管理器

2.下载安装 Anaconda

        (建议安装迅雷下载,同时浏览器添加扩展 “迅雷Chrome支持”)

        https://www.anaconda.com/

        https://repo.anaconda.com/archive(历史版本)

        双击安装包

        换一下路径尽量不要安装在C盘

        安装完了,双击这个运行

        刚开始可能速度有点慢多等等,有更新提示就更新

        有如图所示界面就成功了

        去寻找安装目录,把anaconda3中envs文件夹,安全权限修改全部允许

3.利用 conda 或者 pip 安装 PyTorch 

        确定自己安装的CUDA版本与GPU关系,安装GPU版本 PyTorch

        打开软件的黑窗口

Anaconda命令行,查看虚拟环境
conda env list
创建虚拟环境
conda create -n test python=3.9
开启
conda activate test
关闭
conda deactivate
移除虚拟环境
conda remove -n test --all

指定路径方式
D:
conda create --prefix=D:/Ruanjian/anaconda3/envs/test python=3.9
conda remove --prefix=D:/Ruanjian/anaconda3/envs/test --all
conda activate D:\Ruanjian\anaconda3\envs\test
conda deactivate

        确定自己显卡的算力
        确定自己的可选择CUDA Runtime Version
        确保自己的CUDA Driver 版本>= CUDA Runtime 版本

nvidia-smi

        无法确保版本对应关系,就无脑升级到最新

        安装显卡驱动最新版本
        https://www.nvidia.cn/Download/index.aspx?lang=cn

        按自己的电脑配置来

        打开软件的黑窗口

conda info --env
activate test

        添加清华镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

        输入命令行:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c nvidia  

        应该会下载好久一直等就行

        再等一会......

        下载完成

        验证是否安装成功

import torch
torch.cuda.is_available()

3.配置 PyCharm


        本文章仅供学习交流使用,如有侵权,联系删除

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2023-12-10 16:54:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2023-12-10 16:54:02       100 阅读
  3. 在Django里面运行非项目文件

    2023-12-10 16:54:02       82 阅读
  4. Python语言-面向对象

    2023-12-10 16:54:02       91 阅读

热门阅读

  1. spring bean的生命周期

    2023-12-10 16:54:02       58 阅读
  2. ARM系统软件工程师必备的资源

    2023-12-10 16:54:02       60 阅读
  3. Reactor模型

    2023-12-10 16:54:02       63 阅读
  4. 猜数字游戏与关机程序

    2023-12-10 16:54:02       59 阅读
  5. 安装部署Docker-Compose

    2023-12-10 16:54:02       43 阅读
  6. Linux C++快速入门

    2023-12-10 16:54:02       53 阅读
  7. Vue笔记(二)基本语法

    2023-12-10 16:54:02       49 阅读
  8. 二、安全与风险管理—信息安全治理

    2023-12-10 16:54:02       71 阅读
  9. 信息学奥赛一本通2058

    2023-12-10 16:54:02       53 阅读