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
至此,环境配置完毕