第一步: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-cp36m-win_amd64.whl
·不过,只是用上面的指令,会出现Requirement already satisfied的问题。原因是没有指定安装的目录。
给pytorch指定安装目录(选择刚刚新建的虚拟环境目录)pip install --target=d:\Anaconda3