1. 安装Anaconda
Anaconda清华源下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
在Anaconda Prompt创建虚拟环境:
conda create -n 虚拟环境名(自定义) python=版本号,具体代码:
conda create -n test2 python=3.7
2. 安装CUDA
确定GPU能安装什么版本的CUDA:
首先看自己电脑的显卡:打开任务管理器,点击性能
打开Anaconda的Anaconda Prompt输入命令:nvidia-smi
查看GPU能够支持的最高CUDA版本,我的最高支持到12.0
安装CUDA:
选择能够支持的版本下载cuda,下载地址:https://developer.nvidia.com/cuda-toolkit-archive
我选择的是11.6版本,下载完后就点击下一步安装即可。
安装完成后,打开cmd输入命令nvcc -V
可以查看CUDA版本。
但是!如果显示的是:‘nvcc’ 不是内部或外部命令,也不是可运行的程序或批处理文件。
那首先去检查一下CUDA有没有安装成功。找到安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite。找到下面两个文件
以管理员身份打开cmd,通过cd命令转换到相应的文件位置,输入上图圈中的命令进行测试,若会显示result=pass,则说明安装无误,那么应该是在安装之后没有设置好系统环境变量,重新设置变量之后再进行测试。
添加环境变量:
添加到PATH中:
然后:打开cmd,输入:set PATH=c:
继续输入:echo %PATH%
关闭“cmd"后再次开启cmd,输入echo %PATH%
就会发现新添加的环境变量已经生效,输入nvcc -V
这样就可以正常看到CUDA的信息了:
3.安装pytorch
CUDA版本确定后,pytorch要依赖此CUDA版本,因为pytroch安装跟cuda对应的。
pytorch安装下载的地址:https://download.pytorch.org/whl/torch_stable.html
因为我安装的是11.6的CUDA,安装的python版本是3.10,所以对应安装pytorch:
其中cu代表cuda版本,cp代表python版本,torch代表要安装的pytorch版本。
我们这边是离线安装,会更加稳定更快。
首先要激活安装pytorch的python环境,之后到下载目录下进行安装。
D:\anaconda3\conda_env>pip install "torch-1.13.1+cu116-cp310-cp310-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/
安装成功后测试一下:
import torch
print(torch.__version__) #查看pytorch的版本
print(torch.cuda.is_available()) #查看GPU能否使用,能则返回TRUE
4.安装torchvision
如果要使用计算机视觉,就要安装torchvision。这边依旧采用离线安装,先下载后安装。还是相同的网址:https://download.pytorch.org/whl/torch_stable.html
torch与torchvision也有对应关系,我是1.13.1的torch,3.10的python,所以我选择对应的
0.14.1的torchvision。
找到对应的下载链接:
首先要激活安装pytorch的python环境,之后到下载目录下进行安装。
pip install "torchvision-0.14.1+cu116-cp310-cp310-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/
安装完毕后可以查看一下安装列表conda list
4.安装jupyter
安装anaconda后会自动安装了jupyter notebook,但是这个jupyter是在base环境下的。为了让jupyter使用pytorch,就要在我创建的虚拟环境中下载。
anaconda prompt中激活我创建的ai_env环境,之后依次输入:
conda install ipython
conda install jupyter
打开jupyter后发现无法在Kernel中切换环境,再打开anaconda prompt,在base环境下输入命令conda install nb_conda
,安装nb_conda,之后在jupyter中测试成功。