pytorch基础学习-anaconda方式安装pytorch-gpu版本

前言

这里先引入pytorch-cpu版本的安装方法anaconda方式安装pytorch-cpu，另外关于一些cpu、gpu、ccdn、cuda的理解也参见我上一篇博客1、2点tensorflow基础学习-anaconda方式安装tensorflow（gpu-1.15.0版本

1、创建虚拟环境

打开Anaconda中的 Anaconda Prompt

conda create --name PTGPUPY37 python=3.7

2、激活虚拟环境

输入命令行

activate PTGPUPY37

3、安装pytorch

这里同样可以去pytorch官网确定相应的版本

直接输入对应的命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

4、安装成功后进行测试

这里参考pytorch基础学习-anaconda的方式安装pytorch教程的测试方法

import torch
print(torch.__version__)

两种方式均测试成功，安装成功

注

后面使用可能会出现cuda版本不匹配的情况，可以用以下代码进行测试，如果没有报错，就说明版本对应，不对的话可以自行查相应对应cuda版本进行安装

import torch
torch.zeros(1).cuda()