Anaconda 3-4.2.0
CUDA9.0
python3.5
Pytorch官网地址:https://pytorch.org/get-started/locally/
严格意义上还是需要通过命令行执行,我此次采用的是Anaconda Prompt命令行窗口进行的。
具体步骤如下:
1.去官网查看对应版本的命令
比如我的是
CUDA9.0+python3.5
在官网里就可以查到,然后自动匹配合适的pytorch版本,这个不必担心。
ps.理论上最开始是有自助选择的,但我的浏览器可能有问题,所以自己翻下去去找到的
2.在命令行执行
想在cmd里执行的就找到对应的pip代码
此次是在Anaconda环境下配置,
然后会提示: