Anaconda虚拟环境部署pytorch

一、Anaconda安装并创建虚拟环境

1、官网下载

https://www.anaconda.com/products/individual#Downloads

选择对应版本安装即可

 2、验证是否安装

conda --version

如笔者所示即为正确安装

3、创建虚拟环境

conda create -n 虚拟环境名 python=3.7

python版本自己选择，必须大于3.6 

4、激活虚拟环境

conda activate 虚拟环境名
##激活虚拟环境
conda list
##查看新建环境的包

 

二、Pytorch下载

１、cuda版本检测

win+R输入cmd调出命令行

nvidia-smi

 笔者电脑cuda版本即为11.7

2、pytorch安装

下载三个whl文件

网址：https://download.pytorch.org/whl/

三者版本号与python版本需要对应 

3、安装pytorch文件

  分别用pip下载三个whl文件（到对应下载目录下）

 PS：在安装torchvision时需要额外下载numpy包和pillow包

numpy:https://pypi.tuna.tsinghua.edu.cn/simple/numpy/

pillow:https://pypi.tuna.tsinghua.edu.cn/simple/pillow/

4、检查pytorch是否成功安装 

5、检查pytorch是否可以正常使用

import torch

print(torch.__version__)
print('gpu:',torch.cuda.is_available())

 若torch.cuda.is_available()显示为false，则可能是下列两种误区：

 误区一：直接从官网命令行下载（很慢并且会一直转圈）

 误区二：将condarc改为清华源之后换源下载

无论你是否选择CPU版本，清华源自动给你下载CPU版本，并会直接导致后面验证是否成功安装出错。

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


##附上清华源，下载其他包速度还是很快的