安装pytorch1.9-gpu方法
安装pytorch1.9-gpu方法
1.打开Anaconda Prompt
2.命令行输入:conda create -n PYtorch1.9 python==3.8
此步骤是创建一个conda的环境叫PYtorch1.9基于python3.8
提示后输入y并回车安装相关附带的包
3.命令行输入:conda activate PYtorch1.9
激活并进入PYtorch1.9的环境
4.命令行输入:命令行输入:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
5.验证安装结果:在编译器中设定刚刚安装的环境PYtorch1.9后输入
import torch
a = torch.cuda.is_available()
print(a)
注明:本文为自己的学习总结,供大家交流使用
安装pytorch1.9-gpu方法