form1.cn
Make a little progress every day

windows安装pytorch和nvidia-cuda驱动

02th of November 2021 Linux Command 33

参考视频:https://www.bilibili.com/video/BV1T64y167fK?p=2


0,安装anaconda,创建一个虚拟环境,预装python 3.7


1,下载cuda驱动并安装,下载地址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

请根据实际情况选择,下载后双击安装,共2个G 安装时间10分钟左右,安装完成后在cmd中nvcc -V 测试

要提前检测一下,你的显卡是否支持 cuda  https://blog.csdn.net/Mind_programmonkey/article/details/99688839/ 看这篇文章


2,安装pytorch,打开地址:https://pytorch.org/get-started/locally/,选择后下面会有一行命令:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

cudatoolkit=10.2 此处的版本号可以根据实际情况进行修改,我的就改为了 10.0

安装的时候网络慢会出现中断和错误,这个时候找网络好的环境重新安装一下就可以


3,安装完成之后,写一个demo测试一下,如下:

import torch

import time

from torch import autograd


print(torch.__version__)

print(torch.cuda.is_available())


输出:

1.10.0

True