Window10+GPU配置pytorch
Window10+GPU配置pytorch
环境:Window10 + GPU + CUDA10.0 + cudnn7.5.1 + Anaconda
1、Windows10配置CUDA10.0+cudnn7.5.1
2、Anaconda下载地址,安装地址默认,过程中勾选添加系统环境省去后面手动配置环境
最初下载的是Anaconda2 5.2.0,自带python2.7,需要在其中创建python3.5的环境:
1)在cmd中输入conda create -n py35 python=3.5 ,py35是指你创建的虚拟环境的名字,python=3.5是指python的版本
2)在cmd中输入activate py35 **环境
3)在cmd中输入conda install anaconda ,之前py35中只有原始的包,**环境后在其中安装Anaconda的包
3、pytorch下载地址,根据系统环境下载相应版本
1)cmd切换到下载文件所在目录,输入 pip install torch-1.1.0-cp35-cp35m-win_amd64.whl 进行安装
2)cmd中输入 pip install torchvision
3)cmd中测试pytorch
参考:
win10+1060显卡安装anaconda+CUDA10.1+pytorch+cuDNN+tensorflow-gpu