一、安裝cuda

1、在英偉達官網下載最新版的cuda驅動

https://developer.nvidia.com/zh-cn/cuda-downloads

安裝時都選上就行了，然后一路默認安裝

2、在控制臺輸入nvcc -V查看是否安裝成功

二、安裝pycuda

1、在控制臺中輸入pip install pycuda 安裝pycuda

2、在環境變量中添加cl.exe，（需要提前裝好VS Studio，在此省略了安裝步驟，需要的小伙伴可以自行百度）

3、測試pycuda是否正常運行

import pycuda.driver as drv import pycuda.tools,pycuda.autoinit,numpy import numpy.linalg as la from pycuda.compiler import SourceModulemod = SourceModule(""" __global__ void multiply_them(float *dest, float *a, float *b) {const int i = threadIdx.x;dest[i] = a[i] * b[i]; } """)multiply_them = mod.get_function("multiply_them")a = numpy.random.randn(400).astype(numpy.float32) b = numpy.random.randn(400).astype(numpy.float32)dest = numpy.zeros_like(a) multiply_them(drv.Out(dest), drv.In(a), drv.In(b),block=(400,1,1))print(dest-a*b)

全為0說明GPU計算結果跟CPU一樣

三、安裝GPU版本的pytorch

進入pytorch官網

1、根據剛才下載的cuda和自身環境選擇需要安裝的pytorch版本，選擇后下面會出現一個安裝命令，復制運行即可

2、如果該命令運行失敗的話，一般是由網絡波動，造成下載超時引起的（我裝了三次都是這個原因orz。。）

解決辦法：

1）使用國內 conda 軟件源加速：

$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
$ conda config --set show_channel_urls yes

使用國內pip源的更多辦法

1.臨時設置方法：

可以在使用pip的時候加在最后面加上參數 -i https://pypi.tuna.tsinghua.edu.cn/simple

例如：pip install jieba -i https://pypi.tuna.tsinghua.edu.cn/simple #
jieba 是一個包

2.永久設置方法：

pip install pip -U pip config set global.index-url
https://pypi.tuna.tsinghua.edu.cn/simple

配置完之后就可以像平常一樣安裝包，速度提升幾十倍

例如：pip install jieba

切換為阿里云進行下載

pip install pandas -i http://mirrors.aliyun.com/pypi/simple/
–trusted-host mirrors.aliyun.com pip install pandas -i http://mirrors.aliyun.com/pypi/simple/

阿里云 http://mirrors.aliyun.com/pypi/simple/ 豆瓣(douban)
http://pypi.douban.com/simple/ 清華大學
https://pypi.tuna.tsinghua.edu.cn/simple/ 中國科學技術大學
http://pypi.mirrors.ustc.edu.cn/simple/

2）使用國內源鏡像后對于大文件可能仍然出現下載失敗的情況，對于這些包我們需要采用本地安裝的方式。

通過配置conda config --set show_channel_urls yes在安裝時控制臺會顯示安裝包的下載地址，將下載地址復制下來，使用瀏覽器下載然后通過conda install --offline torchvision-0.4.0-py37_cu100.tar.bz2本地安裝。 （torchvision-0.4.0-py37_cu100.tar.bz2為你需要本地安裝的包，需要先進入其存在的文件夾中才能使用該命令

使用瀏覽器下載的好處是下載錯誤時可以選擇繼續下載，不會一下子報錯。但是有可能會出現網絡波動較強，瀏覽器下載時頻繁錯誤。

解決辦法是，在瀏覽器（Chrome）上安裝使用IDM下載器進行下載（超級好用），效果圖如下：

將這個大文件下載好后進行本地安裝，然后再使用pip命令重新安裝即可，重新安裝時已經安裝完成的文件會跳過，剩下一些小文件搭配國內鏡像很快就能下載好。

（附：清華源pytorch、torchvision下載地址

附：遇到的問題及解決辦法：

1、剛裝好的時候可以加載pycuda，但是無法找到模塊

經過我的分析應該是剛安裝好系統不是很穩定，我重啟了一下電腦解決了問題

2、更改了PATH變量后無反映

修改系統變量后要重啟電腦后才能生效

3、在系統變量中添加某個程序時，要添加他上一級目錄，不用指定到具體哪個程序，系統要調用該程序時會自動變量你添加的所有文件夾。

總結

以上是生活随笔為你收集整理的win10系统配置GPU版本Pytorch的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。