1.#检查cuda的安装情况
nvcc --version
2.将cudnn中的库文件移动到cuda的文件夹中
3.#检查cudnn,跑demo程序,两个PASS

4.#cuda9.0
conda install pytorch1.0.0 torchvision0.2.1 cuda90
5.#添加源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
6.安装pytorch
conda install --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ pytorch=1.0.0 torchvision=0.2.1 cuda90 -c pytorch
win10+cuda9.0+cudnn7.6.5 for cuda9.0+torch1.0
CUDA安装与PyTorch配置
最新推荐文章于 2022-05-08 20:47:04 发布
本文详细介绍了如何检查CUDA的安装状态,包括nvcc版本验证,以及将cuDNN库文件移至CUDA目录的方法。此外,还提供了使用conda安装特定版本PyTorch和TorchVision的步骤,确保与CUDA9.0兼容。最后,通过添加清华镜像源加速安装过程。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
3499

被折叠的 条评论
为什么被折叠?



