windows anaconda python3.7 安装pytorch

已经安装好了anaconda、cuda、cudnn

新建一个虚拟环境

windows anaconda python3.7 安装pytorch_第1张图片

查看自己的cuda 和cudnn版本

cuda:命令行中输入nvcc -V

windows anaconda python3.7 安装pytorch_第2张图片

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 python3.7 安装pytorch_第3张图片

安装skimage

conda install scikit-image

安装cv2

pip install opencv-python

安装torchvision

conda install torchvision -c soumith

 https://pypi.org/project/torchvision/0.1.8/

 

 

你可能感兴趣的:(python,pytorch)