ubuntu+pytorch1.2.0+cuda10.0安装跳的坑

步骤:

  1. conda install cudatoolkit=10.0    先创建cuda 工具,以确保后面的包不会跑到其他版本的cuda
  2. pip install torch==1.2.0
  3. pip install torchvision==0.4.0

服务器自带的镜像torch1.2.0对应cuda9.0,所以要先做步骤1.

猜你喜欢

转载自blog.csdn.net/weixin_45191152/article/details/104496680