GTX 2080Ti + torch0.3.1

2021-02-07  本文已影响0人  热爱生活H_Li

/home/hench/anaconda3/envs/EGAN/lib/python3.5/site-packages/torch/cuda/init.py:95: UserWarning:
Found GPU0 GeForce RTX 2080 Ti which requires CUDA_VERSION >= 9000 for
optimal performance and fast startup time, but your PyTorch was compiled
with CUDA_VERSION 8000. Please install the correct PyTorch binary
using instructions from http://pytorch.org

warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))

本人主机的nvcc版本是10.0的。
pip install torch==0.3.1默认下载的是cuda8.0的,但是2080Ti无法在cuda8中使用,因此需要下载大于cuda9版本的torch。

上一篇下一篇

猜你喜欢

热点阅读