用清华源安装pytorch(gpu)

最新推荐文章于 2024-11-22 09:00:00 发布

weixin_40931921

最新推荐文章于 2024-11-22 09:00:00 发布

阅读量5.5k

点赞数 4

文章标签： python linux

本文链接：https://blog.csdn.net/weixin_40931921/article/details/131623636

版权

本文介绍了如何从清华源下载适用于CUDA10的PyTorchGPU版本1.2.0，并使用conda在虚拟环境中进行离线安装，同时安装torchvision依赖包。安装完成后，通过Python验证CUDA是否可用及版本信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装pytorch-gpu打开清华源下的pytorch库
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
找到自己合适的版本，比如我需要GPU版本的并且支持cuda10，网页ctrl+f快捷搜索，找到文件中
-pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2版本号：1.2.0根据自己需求更改
nightly版本可以看作是开发版，一般不选用，命名格式为 pytorch-版本-python版本-cuda表示gpu版-cudnn版本.tar.bz2大小一般为5、600M下载后，放到合适的盘，接着在刚才创建的那个虚拟环境下，cd到文件目录
输入命令
conda install--offline pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2#文件全称安装完pytorch后，直接输入命令安装完依赖的包：conda install torchvision打开conda中的python进行测试
import torch print（torch.cuda.is_available（））
print（torch.version.cuda）