一、anconda虚拟环境
在这个下面输入指令
创建虚拟环境:conda create -n 名字 python=3.7
切换环境:activate 名字
查看所有环境:conda env list
查看当前环境:python --version
返回默认环境:deactivate 名字 # for Windows
删除环境:conda remove --name 名字 --all
二、NVIDIA驱动安装
官网:https://www.nvidia.cn/geforce/drivers/
查看自己电脑驱动:设备管理器->显示适配器(我的是MX150)
下载好对应版本之后安装(我安装的是最新的驱动,对应的CUDA的是11.4),我是精简安装,最后安装到C盘下面了,别精简安装!!
安装好之后电脑右下角就会有一个NVIDIA设置的图标,到此安装成功
测试:输入nvidia-smi 下面就是你的版本信息
三、anconda的CUDA安装
找到cuda和pytorch对应版本(我的cuda是11.4所以选择的11,cuda我选的10.2)
官网:https://pytorch.org/get-started/previous-versions/
1.添加镜像地址(不是必须,可直接输入安装指令,如下2)
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 --set show_channel_urls yes
删除镜像源
conda config --remove-key channels
2.安装cuda
CUDA 11.0
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
五、anconda的pytorch-gpu安装
1.输入安装指令
pytorch各版本官网:https://pytorch.org/
最后一排就是指令:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
记得添加了镜像源之后要去掉后面的-c pytorch
四、查看安装是否成功
先输入python
mport torch
print(torch.cuda.is_available())
返回True为成功