- 默认nvidia显卡驱动安装成功
- CUDA:
- 查看版本号:https://www.jianshu.com/p/d3b9419a0f89
- CUDA Toolkit Archive : https://developer.nvidia.com/cuda-toolkit-archive
- nvcc -V
- cudnn:
- https://developer.nvidia.com/rdp/cudnn-archive
- 复制bin、include、lib这三个文件夹,粘贴到cuda的路径“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1”下即可。
- (3)将bin所在的目录添加到环境变量 PATH 中,“此电脑”→“高级系统设置”→“环境变量”→“系统变量”→“path”→“编辑”→“新建”加入该路径即可。
- pytorch
- 创建虚拟环境 conda create -n … python=3.6 pip ipython
- https://discuss.pytorch.org/t/install-pytorch-in-windows10/17056/4
pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision
win10 + cuda9.0+pytorch环境配置
最新推荐文章于 2024-01-23 09:16:59 发布