1.conda create -n py37 python=3.7 创建python 3.7的环境
环境名字为py37,名字按自己的想法改
conda create -n your_env_name python=x.x
2.打开创建的环境
conda activate 你的环境名字
如果忘记名字可使用以下命令查看有哪些环境
conda info --envs
3.到pythorch官网,安装cuda对应版本的以及+你想要torch版本的包,进行组合选择
首先使用
nvidia-smi
查看CUDA版本
我的是CUDA Version 11.0
但是我想使用torch-1.10的版本所以我选择CUDA=10.2的,因为毕竟向下兼容,即 11.0兼容10.2
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
然后静静等待安装,出现问号?输入yes
安装好后
4.验证
分次输入一下三串
python
import torch
print(torch.cuda.is_available())
输出结果: Ture
完成-------------------------------------------------------------------------------------------------------------OVER