必备组件:
- Anaconda3-2021.11-Windows-x86_64
- Python3.9
- cuda_11.1.0_456.43_win10
- cudnn-windows-x86_64-8.3.1.22_cuda11.5-archive
- torch-1.10.0+cu113-cp39-cp39-win_amd64
- torchvision-0.11.1+cu113-cp39-cp39-win_amd64
- 编译器,PyCharm
(torch下载的版本要与本地python的版本对应,不然大概率装不上)
步骤:
- 安装Anaconda,目录可以自定,要记得在哪,安装时注意勾选上添加到系统环境变量的选项;
- 安装CUDA,尽量选择默认目录;
- 安装CUDAnn,是个压缩包,解压后复制其中的文件夹到上一步CUDA的安装目录下粘贴覆盖;
- 打开PyCharm,新建一个工程,解释器选择对应的python版本,这里使用的是3.9;
- 工程内新建一个文件,键入如下代码,torch安装之前,import torch应该是标红线报错的;
- 点击左下角Terminal命令窗口,键入:conda create -n pythorch_gpu python=3.9
创建gpu运算环境,建立过程中的[y/n]全部键入y;
- 激活虚拟环境,命令:conda activate pytorch_gpu;
- 激活后,cd到torch与torchversion所在的目录,使用pip install命令进行安装;
- 安装完成后,在当前工程下,更换解释器为刚刚用conda建立的虚拟环境:
- 点击右上角的齿轮,点add,添加新建的解释器(刚刚的torch就安装在这里);
左侧选择conda,右侧点击指定环境,然后找到conda的安装目录,在anaconda>envs目录下找到python.exe, 点击ok>ok; - 最后运行步骤5中的代码,得到如下结果即为安装成功: