一、CUDA安装
CUDA安装如下博客:
(50条消息) 【CUDA】cuda安装 (windows版)_haoweixl的博客-CSDN博客_cuda安装
前提是电脑有NVIDIA独显
在电脑桌面上 点击鼠标右键,出现下图,点击NVIDIA控制面板
点击系统信息,查看显卡型号,驱动程序版本
二、pytorch安装
我在anacaonda里面用清华镜像源安装老是不对,另行他路,参考博客
1、用anaconda 创建一个环境
Win+R启动cmd,在命令提示符内输入以下命令:
conda create –n pytorch python=3.7
activate pytorch
这里一共存在两条指令:
前面一条指令用于创建一个名为pytorch的环境,该环境的python版本为3.7。
后面一条指令用于激活一个名为pytorch的环境。
2、打开pytorch的官方安装方法:
里头有推荐安装的torch 和torchvision版本
我的是CUDA11.1, 推荐安装pytorch==1.8.0 torchvision==0.9.0
# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
下载whl:https://download.pytorch.org/whl/torch_stable.html
cu111表示CUDA的版本是11.1, 若要下载cpu版的,则选择前面是cpu的。
cp38表示python3.8, 记住要安装torch, 再装torchvision。
下载完成后的安装路径:(用的别人的图)
在cmd定位过来后利用文件全名进行安装就行了
3、检验是否安装成功
先输入python,然后输入import torch,如果输入后没有任何报错,没有任何显示那就是成功了,然后再输入torch.cuda.is_available(),返回的是True,那便是完成了整个操作。