PyTorch框架,据说2018.4.25刚刚上架windows,安个玩玩
我的环境:
windows 10
anaconda虚拟环境python3.6
cuda9.1
cudnn7
pytorch 0.4.0
cuda9和cudnn我之前安装tensorflow的时候已经安装过了,这里接着在tensorflow的环境下安装,省的再安装其他模块了
1.whl下载
官网上和别人的教程都是cmd或conda下载安装,我试了试慢的不行,还出错,所以就单独下载whl文件,下载地址http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
这个都是对应我的版本的,其他版本的
2.常规安装
(1)先查看一下虚拟环境,tensorflow-gpu是我3.6的环境,激活一下
(2)切换到whl文件位置安装
(3)successful,测试一下子
(4)最后安装torchversion,我这用的校园网,看外网有点费劲,就让别人帮我下的whl
还是3.6环境下切换到whl文件所在地址,安装
完 美~
原文地址:https://www.cnblogs.com/MC-Curry/p/8994958.html
时间: 2024-10-11 13:53:07