在anaconda中配置pytorch虚拟环境-GPU版

创建环境与一些简单conda指令

• 查看环境: conda env list
• 创建环境: conda create -n 环境名 python=py版本 (-c 镜像地址)
• 删除环境:conda remove -n 环境名 --all
• 激活环境: conda activate 环境名
• 查看环境的库: conda list
• 查看库版本: conda search 库名
• 在环境中安装库: conda install 库名 (-c 镜像地址)
  清华镜像:
  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  北外镜像:
  https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
  阿里镜像:
  https://mirrors.aliyun.com/anaconda/pkgs/main

conda的通道与镜像地址

GPU与CUDA

算力, CUDA Driver Version, CUDA Runtime Version

• 算力-硬件显卡
• CUDA Driver Version-显卡驱动
• CUDA Runtime Version-在pytorch网站上选择的cuda

显卡驱动识别显卡硬件, cuda runtime调用驱动, 驱动调用显卡

• CUDA Driver版本 >= CUDA Runtime版本
• CUDA Runtime Version 支持 显卡对应算力

e.g.RTX2060

1. Google CUDA 查找到RTX2060算力为7.5
2. 确定CUDA Runtime 10.0-10.2, 11.0, 11.1-11.4, 11.5-11.7.1, 11.8, 12满足
3. 看看自己的驱动 CUDA Driver Version为12.0(我的电脑的)cmd中输入nvidia-smi查看
4. 最终确定CUDA Runtime10.0-10.2, 11.0, 11.1...这些小于等于12.0的

安装pytorch

安装pytorch方法1 conda install

安装pytorch方法2 pip install

e.g.

• 1 pytorch官网上找到:
  pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
• 2 修改为镜像加速:
  pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

还是很慢的话可以用迅雷下载

pycharm导入anaconda库

• 1 打开Settings, 选择Project Interpreter, 点击Add interpreter, 选择add local
  
• 2 在System Interpreter中选择添加Anaconda3下的python.exe编译器, 选择OK
  
• 3 选择Anaconda3下的Python编译器, 点击选择Apply, 点击OK