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 **环境

Window10+GPU配置pytorch

3)在cmd中输入conda install anaconda ,之前py35中只有原始的包,**环境后在其中安装Anaconda的包

 

3、pytorch下载地址,根据系统环境下载相应版本

Window10+GPU配置pytorch

1)cmd切换到下载文件所在目录,输入 pip install torch-1.1.0-cp35-cp35m-win_amd64.whl 进行安装

2)cmd中输入 pip install torchvision

Window10+GPU配置pytorch

3)cmd中测试pytorch

Window10+GPU配置pytorch

 

参考:

win10+1060显卡安装anaconda+CUDA10.1+pytorch+cuDNN+tensorflow-gpu