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