windows安装 anaconda3 + CUDA10.2 + pytorch1.5

本机信息

显卡：GTX-1650

CUDA：10.2

1 安装Anaconda

官网下载，安装python3.8版本

安装目录：D:\anaconda\

1.1 换源

用户目录下找到.condarc

  ssl_verify: true
  channels:
  
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    - default
      show_channel_urls: true

1.2 测试

打开anaconda prompt，输入以下内容，显示版本号，说明安装成功

conda --version

1.3 创建自己的虚拟环境

create -n pytorch python=3.6

注：其中pytorch是虚拟环境的名字，python可指定版本号

更换到自己创建的虚拟环境

activate pytorch

1.4 pycharm里面更换解释器

项目中更换解释器为anaconda里的版本即可

2 安装CUDA

CUDA官网找到自己显卡支持的版本

安装即可

2.1 测试

nvcc -V

nvidia-smi

3 安装pytorch

1 找到与自己显卡相符合的CUDA版本，我的显卡为GTX1650，CUDA为10.2

2 在https://pytorch.org/get-started/previous-versions/，找到想要安装的pytorch版本，并且与CUDA版本相同

3 在虚拟环境中进行安装

# CUDA 10.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch

耐心等待~~~

END