linux-pytorch 配置

Posted on Edited on In Views:
linux 上 pytorch 的一些配置

linux-pytorch

基本操作

修改用户密码

1
passwd username

conda

安装 conda

1
2
# 下载
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh
1
2
# 安装
bash Miniconda3-latest-Linux-x86_64.sh
  • 安装完之后好像会写入 .bashrc，导致连接之后默认激活 base 环境
  • 关闭
1
conda config --set auto_activate_base false

新建环境

  • 如下命令是新建一个 python 版本为 3.7、名称为 py37 的环境
1
conda create -n py37 python=3.7

激活/退出环境

1
2
conda activate py37
conda deactivate py37

安装/卸载包

1
2
conda install numpy
conda uninstall numpy

pytorch

  • 查看 cuda 版本
  • 实际安装的版本
1
nvcc -V

  • 与驱动适配的 cuda 版本（不一定是实际安装的版本）
1
nvidia-smi

  • pytorch 官网找对应版本

    • 找不到匹配的版本

    • 使用了这个

1
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch