pytorch安装linux

总结装机必要包

pytorch安装linux

# CUDA 10.0
pip install torch==1.2.0 torchvision==0.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -i https://pypi.tuna.tsinghua.edu.cn/simple

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -i https://pypi.tuna.tsinghua.edu.cn/simple
发布了140 篇原创文章 · 获赞 26 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/CLOUD_J/article/details/104125432
今日推荐