本文档详细介绍了在Windows上已安装Anaconda、CUDA和cuDNN的情况下，如何创建虚拟环境，检查CUDA和cuDNN版本，并安装PyTorch GPU版、skimage、cv2以及torchvision的步骤。确保CUDA版本与PyTorch安装匹配，使用conda和pip进行相关库的安装。
已经安装好了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进行检测：
最低0.47元/天 解锁文章
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
