windows安装 anaconda3 + pytorch

本机信息

显卡：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，输入以下内容，显示版本号，说明安装成功

1	conda --version

1.3 创建自己的虚拟环境

1	create -n pytorch python=3.6

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

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

1	activate pytorch

1.4 pycharm里面更换解释器

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

2 安装CUDA

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

安装即可

2.1 测试

nvcc -V

1	nvidia-smi

3 安装pytorch

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

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

3 在虚拟环境中进行安装

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

耐心等待~~~

路漫漫