赞
踩
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版本。
(2)torchaudio的版本
(3)torchvision的版本
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.验证
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。