pytorch环境

2023-03-25 00:00:00

技术

深度学习,从pytorch开始 pytorch中文手册 https://github.com/zergtant/pytorch-handbook

Linux下

安装Anaconda

在archlinux下

# 完全版
yay -S annaconda
# 本人安装的轻量版
yay -S miniconda

按照提示

If your shell is Bash or a Bourne variant, enable conda for the current user with

    $ source /opt/miniconda/etc/profile.d/conda.sh
You could also add the above line to `~/.bashrc`

or, for all users, enable conda with

    $ sudo ln -s /opt/miniconda/etc/profile.d/conda.sh /etc/profile.d/conda.sh

在终端输入以下命令以启用conda

source /opt/miniconda/etc/profile.d/conda.sh

创建conda环境

# 查看 python 版本
python
# 使用 exit() 或 Ctrl+D 退出 python 环境

# 创建 conda 环境
# ( our_envName 是你定义的环境名,这里使用 pytorch )
# ( x.x 是指定的 python 版本,我这里是 3.10 )
conda create -n your_envName python=x.x

激活环境

# 激活环境,pytorch 是你之前定义的环境名
conda activate pytorch
# 取消激活状态
conda deactivate

安装pytorch

如果需要换源:换清华源 https://blog.csdn.net/weixin_42570192/article/details/124760665

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/

如果需要安装cuda

yay -S cuda

这里不安装cuda,选择cpu,在官网选择对应的安装命令复制粘贴

conda install pytorch torchvision torchaudio cpuonly -c pytorch 

安装好pytorch后

import torch
import torchvision

更多操作命令

https://blog.csdn.net/qq_38870718/article/details/122796306

Windows下

windows在虚拟机里用linux感觉挺卡的,直接用windows算了

安装Anaconda

安装配置

镜像站: https://repo.anaconda.com/archive/

安装后添加环境变量:

D:\anaconda3
D:\anaconda3\Scripts
D:\anaconda3\Library\bin

其中D:\anaconda3是选择的Anaconda安装路径

修改镜像源配置

清华源

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

如果要恢复原来的默认配置

conda config --remove-key channels

创建虚拟环境

创建环境: 命令python可以查看python版本,命令exit()退出python环境

conda create -n pytorch_cpu python=3.10

命令python可以查看python版本,命令exit()退出python环境 激活环境:

activate pytorch

退出激活环境:

deactivate

安装pytorch

官网: https://pytorch.org/ 根据环境选择版本,选择stable-windows-conda-python-none,此配置下的安装命令:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

查看已安装的库

conda list

编辑器配置-vscode

  • 安装python插件,其中包含python相关多个插件
  • 选择编译环境:
    • 随便新建一个python文件
    • 点击右下角python语言模式的右边选择解释器(python.exe)
  • 注意:选择的是带'base'的那个!不是带'pytorch'的那个!它的路径应该为你的Anaconda安装根目录下的python.exe,例如D:\anaconda3\python.exe

至此,环境配置完毕