安装pytorch_d:\a肥仔\a学习\a毕业设计\new-pytorch-chinese-master>pip in

作者：2023面试高手 | 2024-02-28 12:53:45

踩

d:\a肥仔\a学习\a毕业设计\new-pytorch-chinese-master>pip install torch==1.5.0

深度学习环境安装指南

1、安装pytorch
2、安装CUDA

1、安装pytorch

1.1、换源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

   
   
   
   
1
2
3

1.2、如何安装pytorch

pip install torch==1.1.0
pip install torch==0.4.0
1
2

pip install torch1.0.0
pip install torch1.5.0
pip install torch==1.4.0

1.3、查看pytoch是否支持CUDA

命令行输入python进入代码执行界面，输入下面的语句。

torch.cuda.is_available()

 
 
 
 
1

返回True说明支持
以下列出对应版本号：

torch 1.1.0 ->CUDA 9.2
torch 1.2.0 ->CUDA 10.0
torch 1.3.0 ->CUDA 10.1
torch 1.4.0 ->CUDA 10.2
torch 1.5.0 ->CUDA 10.2

 
 
 
 
1
2
3
4
5

2、安装CUDA

暂时不会，建议参考其他博客，安装完成，输入

nvcc -V

 
 
 
 
1

查看CUDA版本号

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