Setup DeepLearning Environment

以下配置基于显卡:NVIDIA GeForce GTX 1650

Anaconda Configuration

  • Download Free download and Skip registration, then choose your platform.
  • 国内镜像源
    1
    https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/    
  • Common Usage
    1
    2
    3
    4
    # remove env
    conda remove -n ENV_NAME --all
    # activate env
    conda activate pytorch

Setup Pytorch Environment

1
2
3
4
5
6
7
8
9
10
11
# setup 
conda create --name=pytorch python=3.7
conda install pytorch==1.7.1 torchvision torchaudio cudatoolkit=11.0 -c pytorch
# for jupyter
conda install nb_conda
conda install nb_conda_kernels

# test
import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

Reference