从0开始的pytorch环境配置
前言:对电脑刷机了,装好了anaconda,但是环境还没配。炼丹必备
gpu相关的
查看显卡:任务管理器查看是否有英伟达显卡
在官网查看自己的显卡是否支持 CUDA 安装
NVIDIA官网地址:https://developer.nvidia.com/cuda-gpus查看自己的显卡驱动版本
命令行输入以下命令
1 | NVIDIA-SMI |
如果没有,请将C:\Program Files\NVIDIA Corporation\NVSMI文件夹放置系统路径的path之中
- 升级显卡驱动程序
去Nvidia官网地址 https://developer.nvidia.com/cuda-downloads 按系统版本,选择下载在线升级包, 下载完后,选择自定义先升级显卡驱动,如图:
- 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 | %进入python环境 |
1 | dir(torch) |
5.jupyter编辑器
Anaconda Prompt
输入指令
1 | conda activate mytorch |
6.查看和删除
1 | 环境的查看 |
8.一些镜像源
1 | pip install -r requirements.txt |
- Could not find a version that satisfies the requirement
1
python -m pip install --upgrade pip
1
2
3
4
5
6
7
8
9pip install <库名> 安装
pip uninstall <库名> 卸载
pip install -U <库名> 更新
pip download <库名> 只下载,不安装
pip show <库名> 展示这个库的信息
pip search <库名> 搜索关键字
pip list 列出安装的库
查看用conda安装的包:
conda list