前言
这里先引入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__)
两种方式均测试成功,安装成功