一.查看服务器的cuda版本

使用如下命令:

nvcc -V

博主的cuda版本为10.1

二.安装对应版本的pytorch

进入pytorch官网 https://pytorch.org/ 选择对应的版本

输入提示的安装命令,安装过慢可以使用镜像源

pip install torch torchvision -i  https://pypi.tuna.tsinghua.edu.cn/simple

三.检查是否安装成功

>>> import torch
>>> torch.__version__

显示pytorch版本为1.3.0,安装成功!

GitHub 加速计划 / li / linux-dash
10.39 K
1.2 K
下载
A beautiful web dashboard for Linux
最近提交(Master分支:2 个月前 )
186a802e added ecosystem file for PM2 4 年前
5def40a3 Add host customization support for the NodeJS version 4 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐