好久没有更新pytorch,这次更新做个笔记
1.安装CUDA
设置环境变量后输入nvcc -V
C:\Users\WQBin>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c)2005-2019NVIDIA Corporation
Built on Fri_Feb__8_19:08:26_Pacific_Standard_Time_2019
Cuda compilation tools, release10.1, V10.1.105
2.安装pytorch
PyTorch on Windows only supports Python 3.x
2.1解决慢到无法装的问题
最有效方法是添加镜像源,常见的清华或中科大。先查看是否已经安装相关镜像源,windows系统在cmd窗口中执行命令:
conda config --show
使用【conda config --add channels 】添加清华数据源
channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/
- https://mirrors.tuna.tsingh