当前位置:   article > 正文

win10+pytorch+cuda11.7_cuda11.7对应的torch版本

cuda11.7对应的torch版本

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.验证

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/173358
推荐阅读
相关标签
  

闽ICP备14008679号