已经安装好了anaconda、cuda、cudnn
新建一个虚拟环境

查看自己的cuda 和cudnn版本
cuda:命令行中输入nvcc -V

cudnn:windows中打开 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h
cuda 9.0,cudnn 7.6.1
安装pytorch gpu
参考:https://www.cnblogs.com/GrPhoenix/p/10324172.html
安装 PyTorch,注意 cuda 的版本要和你自己的系统保持一致,比如在 CUDA 9.0 上安装:
conda install -c pytorch pytorch-nightly cuda90
安装 fastai:
conda install -c fastai fastai
进行检测:

本文档详细介绍了在Windows上已安装Anaconda、CUDA和cuDNN的情况下,如何创建虚拟环境,检查CUDA和cuDNN版本,并安装PyTorch GPU版、skimage、cv2以及torchvision的步骤。确保CUDA版本与PyTorch安装匹配,使用conda和pip进行相关库的安装。
最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



