cuda+pytorch环境安装

1. 安装cuda

  1. nvidia-smi查看GPU信息,显示的CUDA Version是可以支持的最高版本cuda
  2. 官网安装相应版本的cuda
  3. 安装后重新打开终端,nvcc --version,输出cud版本

2. 安装conda/miniconda

2.1 安装

anaconda官网按指示安装,使用onda --version看conda版本

  • macos
  • win 下载exe安装

2.2 常用conda命令

2.2.1 conda配置信息

  • conda info 查看conda信息,里面有配置文件位置.condarc
  • conda config --show [xx] 显示某项
  • 一般配置一些channel和proxy
conda config --show channels
conda config --add channels conda-forge
conda config --remove channels conda-forge
  • conda镜像源
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

2.2.2 环境管理

conda create --name xxx python=3.12
# 移除
conda remove --name xxx --all
conda remove --name xxx package_name
# 导出
conda env export --name xxx > xxx.yml
# 还原
conda env create -f xxx.yml

2.2.3 conda包管理

conda list                # 查看当前环境中的包
conda list package_name   # 查看当前环境中是否安装了
conda list package_name*  #模糊查询
conda search package_name # 在anaconda repository中查询目标包
conda update package_name

2.2.4 conda install & pip install

  • 先conda再pip
  • pip是给指定的环境安装的

3. 安装PyTorch

nvcc -V查看cuda版本
torch官网
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

posted @ 2023-02-02 14:08  不要肥宅  阅读(61)  评论(0编辑  收藏  举报