品牌营销策划书_网站建设哪个最好_百度还原_活动推广文案

创建环境

conda create -n 环境名称 python=3.8

激活新建的环境

conda activate 环境名称

安装需要的平时的库（不包括pytorch）

pip install numpy

自己根据需要安装

查看电脑的显卡驱动版本+安装cuda

在cmd命令窗口中输入nvidia-smi，查看CUDA Version
根据这个里面安装好CUDA 全网最详细的安装pytorch GPU方法，一次安装成功！！包括安装失败后的处理方法！
，终端输入nvcc -V 查看是否安装好。

安装pytorch

在 pytorch网站 根据自己电脑情况，选择pytorch。

复制pip语句到终端，进行安装。

测试pytorch是否用gpu

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True，则安装无误

终端输出为True，就安装好了。