pytorch安装

安装好Anaconda

打开Anaconda prompt 

conda create -n pytorch python=3.6

激活创建的环境

conda activate pytorch

查看当前的环境中的包 有哪些

pip list

进入pytorch官网

 这里需要查看自己的显卡支持什么样的算力

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装完成后验证是否安装完毕：返回true

pip list
python
import torch
torch.cuda.is_available