从0开始的pytorch环境配置

从0开始的pytorch环境配置

前言:对电脑刷机了,装好了anaconda,但是环境还没配。炼丹必备

gpu相关的

  1. 查看显卡:任务管理器查看是否有英伟达显卡

  2. 在官网查看自己的显卡是否支持 CUDA 安装
    NVIDIA官网地址:https://developer.nvidia.com/cuda-gpus

  3. 查看自己的显卡驱动版本
    命令行输入以下命令

1
NVIDIA-SMI

如果没有,请将C:\Program Files\NVIDIA Corporation\NVSMI文件夹放置系统路径的path之中

  1. 升级显卡驱动程序

去Nvidia官网地址 https://developer.nvidia.com/cuda-downloads 按系统版本,选择下载在线升级包, 下载完后,选择自定义先升级显卡驱动,如图:

img

img

  1. CUDA安装是否成功
1
nvcc -V

python环境安装

1.定义环境

1
conda create -n mytorch python==3.9

2.运行环境

1
conda activate mytorch

3.安装https://pytorch.org/get-started/locally/
pytorch:

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
1
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia -i https://pypi.tuna.tsinghua.edu.cn/simple

4.测试一下

1
2
3
4
%进入python环境 
python
import torch
torch.cuda.is_available()
1
2
dir(torch)
help(torch.cuda.is_available)

5.jupyter编辑器
Anaconda Prompt
输入指令 

1
2
3
4
conda activate mytorch
jupyter notebook
#打开jupyter后
import torch

6.查看和删除

1
2
3
4
5
环境的查看
Anaconda Prompt
conda info -e
环境的删除
conda remove --name yolo5 --all

8.一些镜像源

1
2
3
4
5
6
7
8
9
10
pip install -r requirements.txt
pip install opencv-python
pip install tensorflow-cpu==2.3.0
-i https://mirror.baidu.com/pypi/simple
-i https://pypi.tuna.tsinghua.edu.cn/simple
-i http://pypi.douban.com/simple/
阿里云:http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学:http://pypi.hustunique.com/
山东理工大学:http://pypi.sdutlinux.org/
  1. Could not find a version that satisfies the requirement
    1
    python -m pip install --upgrade pip
  2. 1
    2
    3
    4
    5
    6
    7
    8
    9
    pip install <库名>			安装
    pip uninstall <库名> 卸载
    pip install -U <库名> 更新
    pip download <库名> 只下载,不安装
    pip show <库名> 展示这个库的信息
    pip search <库名> 搜索关键字
    pip list 列出安装的库
    查看用conda安装的包:
    conda list
-------------未完待续欢迎下次光临-------------
谢谢你想打赏的心~