请先安装Anaconda和CUDA 10.0。
- 配置国内源
# 配置国内源,方便安装Numpy,Matplotlib等
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/
# 配置国内源,安装PyTorch用
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 显示源地址
conda config --set show_channel_urls yes
- 安装PyTorch
# 安装PyTorch,要使用国内源请去掉-c pytorch这个参数!!
conda install pytorch torchvision cudatoolkit=10.0
- 安装常用库
pip install numpy matplotlib pillow pandas