win10基于Anaconda3的PyTorch1.4.0 GPU10.1版本的安装

最新推荐文章于 2024-08-12 18:30:09 发布

ITxiaolizi

最新推荐文章于 2024-08-12 18:30:09 发布

阅读量1.1k

点赞数 1

文章标签：机器学习深度学习 pytorch 神经网络

本文链接：https://blog.csdn.net/ITxiaolizi/article/details/105253609

版权

下载安装英伟达CUDA10.1 https://developer.nvidia.com/cuda-10.1-download-archive-base

搜索栏输入cmd，以管理员方式打开cmd
更换清华大学开源下载库，pytorch版本为1.4.0
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安装 PyTorch
conda install pytorch torchvision cudatoolkit=10.1
测试是否安装成功
打开Anaconda Prompt (Anaconda3)
输入 python
1.import torch
2.print(torch.__version~~删除线格式~~ __)
3.print(torch.version.cuda)
4.print(torch.cuda.is_available())