虚拟环境命令行安装torch+cuda_cuda torch安装指令

作者：2023面试高手 | 2024-04-06 09:03:27

踩

cuda torch安装指令

win+r输入cmd 回车输入nvidia-smi查看驱动版本为12.0，所以下载的cuda版本不超过12.0

打开Anaconda Prompt

1.创建名为DL的虚拟环境：conda create -n DL

2.进入名为DL的虚拟环境：conda activate DL

3.查看环境中有哪些包：pip list

4.下载命令在官网Previous PyTorch Versions | PyTorch中复制

本人下载的cuda为11.3版本，末尾可以加上清华镜像。

-i https://pypi.tuna.tsinghua.edu.cn/simple

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/370966