今天安裝pycuda
先安裝 nvidia 的 cuda toolkit. 到 nvidia 官網下載安裝包。cuda_11.1.1_456.81_win10.exe
通過下面命令確認 cuda 安裝成功了：
C:\Users\allen>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29190527_0
然后安裝了visual studio community 2019, 微軟網站下載即可， 其中需要選上 "使用 C++ 的桌面開發"。因為安裝 pycuda 的過程中需要用到 Visual C++ 14
最后pip 安裝 pycuda:
C:\Users\allen>pip install pycuda -i https://pypi.tuna.tsinghua.edu.cn/simple
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting pycuda
Using cached https://pypi.tuna.tsinghua.edu.cn/packages/46/61/47d3235a4c13eec5a5f03594ddb268f4858734e02980afbcd806e6242fa5/pycuda-2020.1.tar.gz (1.6 MB)
Requirement already satisfied: pytools>=2011.2 in c:\python37\lib\site-packages (from pycuda) (2020.4.3)
Requirement already satisfied: decorator>=3.2.0 in c:\python37\lib\site-packages (from pycuda) (4.4.2)
Requirement already satisfied: appdirs>=1.4.0 in c:\python37\lib\site-packages (from pycuda) (1.4.4)
Requirement already satisfied: mako in c:\python37\lib\site-packages (from pycuda) (1.1.3)
Requirement already satisfied: numpy>=1.6.0 in c:\python37\lib\site-packages (from pytools>=2011.2->pycuda) (1.19.0)
Requirement already satisfied: six>=1.8.0 in c:\python37\lib\site-packages (from pytools>=2011.2->pycuda) (1.15.0)
Requirement already satisfied: MarkupSafe>=0.9.2 in c:\python37\lib\site-packages (from mako->pycuda) (1.1.1)
Building wheels for collected packages: pycuda
Building wheel for pycuda (setup.py) ... done
Created wheel for pycuda: filename=pycuda-2020.1-cp37-cp37m-win_amd64.whl size=366928 sha256=e7bb699608ec79e65cae27e361d659c7038a30ee261f3d1459f99b04d24a6eff
Stored in directory: c:\users\allen\appdata\local\pip\cache\wheels\b3\ce\35\9844bc8ade7f164b636a9e5a1f559f38c74960ae55531fedb0
Successfully built pycuda
Installing collected packages: pycuda
Successfully installed pycuda-2020.1
WARNING: You are using pip version 20.1.1; however, version 20.2.4 is available.
You should consider upgrading via the 'c:\python37\python.exe -m pip install --upgrade pip' command.
網上找個 python 調用 cuda 的例子。執行。如果遇到：nvcc fatal : Cannot find compiler 'cl.exe' in PATH”
就到 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community 下面搜索 cl.exe, 然后把路徑添加到 PATH 中。