win10+pytorch+cuda11.7

1.打开Anaconda Prompt,创建一个新的虚拟环境

语法:conda create -n [虚拟环境名称] python=[版本号]

conda create -n torchGPU python=3.7

2.激活创建的虚拟环境

语法:activate [虚拟环境名字]

或conda activate [虚拟环境名字]

activate torchGPU

3.查看合适的版本(可以跳过这步)

进下面这个链接查看版本,注意pytorch和torchvision版本要对应(自行搜索)

https://download.pytorch.org/whl/cu117/torch_stable.html

(1)torch的版本

torch版本为1.13.0+cuda为11.7+python3.7+win系统

cu就是代表cuda版本,cp代表python版本。

win10+pytorch+cuda11.7_第1张图片

(2)torchaudio的版本

win10+pytorch+cuda11.7_第2张图片

(3)torchvision的版本

win10+pytorch+cuda11.7_第3张图片

4.安装

语法:pip3 install torch==[pytorch版本号]+cu[cuda版本号] torchvision==[torchvision版本号]+cu[cuda版本号] torchaudio==[torchaudio版本号]+cu[cuda版本号] -f https://download.pytorch.org/whl/cu[cuda版本号]/torch_stable.html

pip3 install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0+cu117 -f https://download.pytorch.org/whl/cu117/torch_stable.html

5.验证

win10+pytorch+cuda11.7_第4张图片

你可能感兴趣的:(pytorch,python,深度学习)