第一步:CUDA版本查看
使用win+r,在输入cmd,在小黑窗口中输入nvcc -V
第二步:创建一个新的Python虚拟运行环境
- 在小黑窗口输入conda create -n 名称(如pytorch-gpu) python==3.6.2 然后回车。
- 激活刚刚创建的虚拟运行坏境 activate pytorch-gpu
- 在新建的虚拟环境中安装pytorch,使用 pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36